Model Training Python Packages

pictograph

Complete python SDK, CLI, and API reference for integrating with the Pictograph.io platform.

hub-sdk

Retired Python SDK for the shut-down Ultralytics HUB API, kept for reference; build new integrations on the Ultralytics Platform REST API

dataexcept

A comprehensive Python library providing structured, hierarchical exception classes for data science, data engineering, and ML pipeline workflows. Features 40+ specialized exceptions, logging helpers, and production-ready error handling for robust data processing systems.

ultralytics-yolov5

Ultralytics YOLOv5 in PyTorch for object detection, instance segmentation, classification, training, and export.

maarg

Lightweight experiment tracking for Python with zero boilerplate — automatically capture inputs, metrics, artifacts, execution time, and failures.

weightslab

Monitoring platform for ML teams building real‑world AI at scale.

fotonet

Compact NMS-free object detection in PyTorch with resumable training, COCO evaluation, Python APIs, and ONNX or TorchScript export.

delve

PyTorch model training and layer saturation monitor

cotorra

A configurable trainer for generative event models

fast-axolotl

High-performance Rust extensions for Axolotl (no OOM for large datasets) - drop-in acceleration for existing installations.

mlflow-cratedb

MLflow adapter for CrateDB.

dataflex

[NeurIPS 2026] DataFlex: A Unified Benchmark and Evaluation Platform for Data-Centric Training of Large Language Models

silver-torch

Inspectable PyTorch training with visual layer shapes, activations, sparsity, gradients, and topology.

visagesnap

Easy-to-Use library to recognize faces, train the model, provide predictions

profine

Profine automatically profiles and optimizes PyTorch training jobs on real GPUs, delivering measurable speedups and lower GPU costs before teams waste days tuning configs by hand.

drillvision

This repository contains code for training a convolutional neural network (CNN) model to classify images.

gaama

Experiments with Model Training, Deployment & Monitoring

piepline

Neural networks training pipeline based on PyTorch

flatbuild

Build language models from scratch. Simple, modular, and reproducible training for conversational AI.

skrobot

skrobot is a Python module for designing, running and tracking Machine Learning experiments / tasks. It is built on top of scikit-learn framework.

xlmtec

xlmtec is a powerful, modular, and interactive command-line tool for fine-tuning Large Language Models (LLMs). Multiple Techniques: Support for LoRA, QLoRA, and Prompt Tuning. Benchmarking: Built-in evaluation with ROUGE and other metrics.

mlfcrafter

ML Pipeline Automation Tool - Chain together data processing, model training, and deployment with minimal code. Build production-ready ML workflows in minutes, not hours.

yolov5-inference

Ultralytics YOLOv5 in PyTorch for object detection, instance segmentation, classification, training, and export.

grainz

The embedding lifecycle, end to end — store, search, cache, and track embeddings behind one consistent surface.