Training Python Packages

swesmith

[NeurIPS 2025 D&B Spotlight] Scaling Data for SWE-agents

torchao

PyTorch native quantization for training and inference

optimum

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

sagemaker-training

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

torchft-nightly

Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)

swanlab

⚡️SwanLab - an open-source, modern-design AI training tracking and visualization tool. Supports Cloud / Self-hosted use. Integrated with PyTorch / Transformers / verl / LLaMA Factory / ms-swift / Ultralytics / MMEngine / Keras etc.

dstack

A unified orchestration layer for heterogeneous AI compute. It standardizes how to manage compute and run training and inference on GPU clouds, Kubernetes, VMs, or bare-metal clusters.

tokenicer

A (nicer) tokenizer you want to use for model inference and training: with all known peventable gotchas normalized or auto-fixed.

dsoxlab

Turn declarative exercises into reproducible, runnable and verifiable lab environments

hypernix

Ai training toolkit

mlx-commander

Orthodox ML model training manager for Apple Silicon

mlx-lm-lora

Train Large Language Models on MLX.

epochix

Visual storytelling for deep learning training runs.

split-folders

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

datasetrising

Toolchain for creating custom datasets and training Stable Diffusion (1.x, 2.x, XL) models and LoRAs

awex

A high-performance RL training-inference weight synchronization framework, designed to enable second-level parameter updates from training to inference in RL workflows

backpropagate

Headless LLM fine-tuning in 3 lines — smart defaults, VRAM-aware batch sizing, multi-run SLAO, GGUF export for Ollama.

training-hub

An algorithm-focused interface for common llm training, continual learning, and reinforcement learning techniques

trainspotter

Automatic diagnosis of training runs from their logs: divergence, loss spikes, overfitting, LR and throughput problems.

avalanche-lib

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

neptune-sklearn

Experiment tracking for scikit-learn. 🧩 Log, organize, visualize and compare model metrics, parameters, dataset versions, and more.

runhouse

Distribute and run AI workloads on Kubernetes magically in Python, like PyTorch for ML infra.

dbasim

Oracle DBA break-fix simulator: practise real incidents on your own Oracle Database Free

deepepochs

Pytorch模型训练工具