Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code.

SSD-based object and text detection with Keras, SSD, DSOD, TextBoxes, SegLink, TextBoxes++, CRNN

Distributed training (multi-node) of a Transformer model

🎯 Gradient Accumulation for TensorFlow 2

Gradient accumulation for Keras

TorchHandle makes your PyTorch development more efficient and make you use PyTorch more comfortable

Gradient accumulation on tf.estimator

A simple implementation of Multi-passage BERT

🎯 Production-ready implementation of video prediction models using PyTorch. Features Enhanced ConvLSTM with temporal attention, PredRNN with spatiotemporal memory, and Transformer-based architecture.

tensorflow2-keras gradient accumulation

This project aims to help people implement tensorflow model pipelines quickly for different nlp tasks.

Gradient Accumulation with Tensorflow2.x

Comprehensive PyTorch Lightning framework featuring 20+ educational notebooks, advanced ML patterns, and production-ready workflows. Covers vision, NLP, tabular, and time series domains with distributed training, mixed precision, custom loops, and deployment pipelines. Complete with synthetic data generators and testing.

Research code implementing the "Attention Is All You Need" architecture. Engineers a stable training loop for a 163M LLM using reduced-precision techniques on free-tier compute.

Implementation of Gradient Accumulation for low-memory language modelling transformer fine tuning.

Classifying images of flowers into 17 categories using EfficientNet-B0 and PyTorch.

AlzMRI-Net: Classify Alzheimer's stages from MRI scans.