Machine Learning Engineering Open Book

🚀 Accelerate inference and training of 🤗 Transformers, Diffusers, TIMM and Sentence Transformers with easy to use hardware optimization tools

OneTrainer is a one-stop solution for all your stable diffusion training needs.

PyTorch native quantization and sparsity for training and inference

Avalanche: an End-to-End Library for Continual Learning based on PyTorch.

dstack is a lightweight, open-source alternative to Kubernetes & Slurm, simplifying AI container orchestration with multi-cloud & on-prem support. It natively supports NVIDIA, AMD, TPU, and Intel accelerators.

Streamlining reinforcement learning with RLOps. State-of-the-art RL algorithms and tools.

Train Tesseract LSTM with make

A unified end-to-end machine intelligence platform

Train machine learning models within a 🐳 Docker container using 🧠 Amazon SageMaker.

🗂 Split folders with files (i.e. images) into training, validation and test (dataset) folders

simpleT5 is built on top of PyTorch-lightning⚡️ and Transformers🤗 that lets you quickly train your T5 models.

Guideline following Large Language Model for Information Extraction

OTOv1-v3, NeurIPS, ICLR, TMLR, DNN Training, Compression, Structured Pruning, Erasing Operators, CNN, Diffusion, LLM

Unsupervised Deep Homography: A Fast and Robust Homography Estimation Model

The pure and clear PyTorch Distributed Training Framework.

PyTorch per step fault tolerance (actively under development)

A straightforward method for training your LLM, from downloading data to generating text.

Simple Pose: Rethinking and Improving a Bottom-up Approach for Multi-Person Pose Estimation

DAWNBench: An End-to-End Deep Learning Benchmark and Competition