Training, Evaluation & Optimization

Training frameworks, fine-tuning, evaluation, observability, and optimization workflows.

87Projects8Subcategory62Tags
TrackedUnavailableArchivedInactive

Training frameworks and distributed training systems.

AXLearn

An extensible deep learning library built on JAX/XLA, designed for developing, training and deploying large-scale models.

Stars --Last Commit --

ClearML

ClearML is an open-source MLOps platform providing experiment tracking, data management, pipelines and model serving.

Stars --Last Commit --

Colossal-AI

Discover Colossal-AI: an open-source solution for efficient large-scale training and inference, featuring advanced parallelism and memory management for optimal performance.

Stars --Last Commit --

DeepEP

An efficient expert-parallel communication library that provides low-overhead communication primitives for large-scale distributed training.

Stars --Last Commit --

DeepSpeed

A high-performance library for training and inference that dramatically speeds up large-scale deep learning while reducing cost.

Stars --Last Commit --

DLRover

DLRover is an automatic distributed deep learning system that provides elastic scheduling, flash checkpointing and auto-scaling to simplify large-scale model training on Kubernetes and Ray.

Stars --Last Commit --

EasyR1

EasyR1 is an efficient, scalable RL training framework for multimodal models, based on veRL and optimized for large-model training.

Stars --Last Commit --

Gymnasium

An API standard for single-agent reinforcement learning environments, with popular reference environments and related utilities (formerly Gym).

Stars --Last Commit --

JAX

High-performance Python library for accelerator-oriented array computation and composable program transformations.

Stars --Last Commit --

LightGBM

A fast, distributed, high-performance gradient boosting framework for decision tree algorithms, widely used for ranking, classification, and large-scale ML tasks.

Stars --Last Commit --

LitGPT

A high-performance, engineering-focused LLM toolkit that provides end-to-end recipes and practical tutorials for training and deploying large models.

Stars --Last Commit --

MaxText

A high-performance, highly scalable open-source LLM library and reference implementation built with Python and JAX, targeting Google Cloud TPUs and GPUs.

Stars --Last Commit --

Megatron-LM

Reference implementation from NVIDIA for large-scale model training and inference with distributed optimizations.

Stars --Last Commit --

MONAI

An AI toolkit for healthcare imaging focused on deep learning workflows for medical image processing and analysis.

Stars --Last Commit --

nanoGPT

A minimal, fast repository for training and fine-tuning medium-sized GPT models, suitable for teaching and experiments.

Stars --Last Commit --

NeMo

NVIDIA's NeMo framework for speech, TTS, multimodal and LLM training & fine-tuning.

Stars --Last Commit --

NeMo RL

NeMo RL is a scalable post-training reinforcement learning library for large models, supporting high-performance distributed training and multiple backends.

Stars --Last Commit --

OpenEnv — Agentic Execution Environments

An end-to-end framework for creating, deploying and using isolated execution environments, aimed at agentic RL training and environment development.

Stars --Last Commit --

PyTorch

An open-source deep learning framework for fast, flexible research and production, featuring dynamic computation graphs and strong GPU acceleration.

Stars --Last Commit --

PyTorch Lightning

PyTorch Lightning is an open-source framework that streamlines PyTorch training, enabling efficient model development, training, and deployment.

Stars --Last Commit --

RLinf

RLinf is a flexible and scalable open-source RL infrastructure designed for Embodied and Agentic AI, supporting PPO, GRPO, SAC and more, with seamless scaling to large GPU clusters.

Stars --Last Commit --

ROLL

Reinforcement Learning Optimization platform for large-scale training and pipelines.

Stars --Last Commit --

SkyRL

A modular full-stack reinforcement learning (RL) library for large language models (LLMs), designed for long-horizon, real-world tasks.

Stars --Last Commit --

TensorFlow

Google's open-source end-to-end machine learning platform for building and training deep learning models.

Stars --Last Commit --

TorchTitan

A PyTorch-native platform for generative model pretraining and distributed optimization.

Stars --Last Commit --

verl

A reinforcement learning training framework for large models, designed for scalable RLHF and agent training.

Stars --Last Commit --

xLLM

xLLM is an open-source framework for vision-language models, providing tools and documentation for training and inference.

Stars --Last Commit --

SFT, RLHF, preference optimization, and alignment.

AReaL

A fully asynchronous reinforcement learning system for large reasoning and agentic models that emphasizes scalability and reproducibility.

Stars --Last Commit --

Axolotl

A free and open-source LLM post-training and fine-tuning framework that supports multiple models, training methods, and distributed optimizations.

Stars --Last Commit --

Heretic

Heretic is a fully automated tool that removes censorship (aka "safety alignment") from transformer-based language models without expensive post-training. It combines an advanced implementation of directional ablation, also known as "abliteration," with a TPE-based parameter optimizer powered by Optuna to automatically find high-quality ablation parameters by co-minimizing refusals and KL divergence from the original model.

Stars --Last Commit --

LLaMA Factory

A comprehensive framework for fine-tuning LLaMA models with multiple training methods, efficient algorithms, and easy-to-use interface for both research and production environments.

Stars --Last Commit --

LMFlow

An extensible, convenient, and efficient toolbox for fine-tuning and inference of large foundation models.

Stars --Last Commit --

MLX-VLM

A local-first toolkit for inference and fine-tuning of vision-language and omni models using MLX, optimized for macOS and general hardware.

Stars --Last Commit --

MS-SWIFT

SWIFT from ModelScope: a scalable, lightweight infrastructure for fine-tuning, evaluating and deploying large and multimodal models, with training, quantization and inference acceleration support.

Stars --Last Commit --

OpenRLHF

An easy-to-use, high-performance open-source RLHF framework built on Ray, vLLM and DeepSpeed, supporting distributed and hybrid-engine training.

Stars --Last Commit --

PEFT

State-of-the-art parameter-efficient fine-tuning methods for large language models, enabling adapter-based training with minimal GPU resources.

Stars --Last Commit --

Torchtune

A PyTorch-native post-training and fine-tuning toolkit providing reusable recipes, optimizations, and quantization support for LLM training and evaluation.

Stars --Last Commit --

Transformer Lab

Explore Transformer Lab, the open-source app for downloading and fine-tuning large models locally or in the cloud with powerful tools and multi-engine support.

Stars --Last Commit --

TRL

TRL is an open-source toolkit from Hugging Face for reinforcement learning training on transformer models.

Stars --Last Commit --

Tunix

Tunix is a JAX-native post-training library for LLMs providing efficient fine-tuning, RL training, and distillation tools.

Stars --Last Commit --

Unsloth

High-performance toolkit for fine-tuning and reinforcement learning of large models, with memory-efficient kernels and wide model support.

Stars --Last Commit --

Experiment tracking, model ops, and training pipelines.

Google Research

Google Research aggregates open-source research code and datasets from Google, covering machine learning, vision, NLP and other research areas.

Stars --Last Commit --

Skypilot

Skypilot is an open-source tool to automate distributed training and inference across cloud and on-premises clusters, simplifying resource provisioning and environment setup.

Stars --Last Commit --

Slurm

Slurm is an open-source cluster resource management and job scheduling system that is simple, scalable, portable, fault-tolerant, and interconnect agnostic, widely used in high-performance computing and AI training clusters.

Stars --Last Commit --

SwanLab

SwanLab is an open-source, modern training tracking and visualization tool that supports cloud and self-hosted deployment.

Stars --Last Commit --

Weights & Biases (W&B)

A machine learning development and observability platform for tracking experiments, managing models and artifacts, and visualizing results across the ML lifecycle.

Stars --Last Commit --

ZenML

A unified MLOps framework to develop, evaluate and deploy everything from classical models to multi-agent AI systems.

Stars --Last Commit --

Evaluation frameworks, benchmark suites, and datasets.

Agenta

Agenta is an open-source LLMOps platform that combines prompt management, evaluation, and observability to help teams ship reliable LLM applications faster.

Stars --Last Commit --

ClawBench

ClawBench is an open-source benchmark for evaluating browser AI agents on everyday tasks across live production websites, with isolated runs and five-layer execution traces.

Stars --Last Commit --

DeepEval

DeepEval is an open-source LLM evaluation framework that provides modular metrics and tooling for testing LLM systems and RAG pipelines.

Stars --Last Commit --

DeepTeam

An open-source framework for red-teaming large language models and LLM systems, focused on security and robustness evaluation.

Stars --Last Commit --

Dingo

A tool for automated data quality evaluation that combines rule-based and model-based assessments.

Stars --Last Commit --

EasyEdit

An easy-to-use knowledge editing framework providing multiple editing methods, evaluation metrics and datasets; supports LLMs and some multimodal editing scenarios.

Stars --Last Commit --

Evidently

An open-source framework for evaluating, testing, and monitoring ML and LLM systems from experiments to production.

Stars --Last Commit --

fuck-u-code

A static analysis tool that assesses codebase 'legacy-mess' and generates readable Markdown reports.

Stars --Last Commit --

Giskard OSS

An open-source evaluation and testing framework to detect performance, bias, and security issues in AI systems.

Stars --Last Commit --

HELM

Holistic Evaluation of Language Models (HELM) from Stanford CRFM: an open framework for reproducible, transparent model evaluation and benchmark management.

Stars --Last Commit --

Inspector

Inspector is a visual testing tool for MCP (Model Context Protocol) servers that helps developers validate and visualize server behavior and responses.

Stars --Last Commit --

Keploy

A developer-centric API and integration testing tool that auto-generates tests and data mocks from real traffic, supporting record-and-replay of API calls, database operations, and streaming events.

Stars --Last Commit --

LightEval

A lightweight toolkit from Hugging Face for fast, flexible LLM evaluation across multiple backends.

Stars --Last Commit --

LiveBench

LiveBench is a contamination-aware, objective LLM benchmark suite that provides reproducible question sets, automatic scoring, and an online leaderboard.

Stars --Last Commit --

lm-evaluation-harness

The Language Model Evaluation Harness is a framework for large-scale, reproducible evaluation of generative language models across many tasks and datasets.

Stars --Last Commit --

OpenCompass

A one-stop platform for evaluating large models, providing benchmarks, evaluation toolkits and leaderboards to reproduce and compare model capabilities.

Stars --Last Commit --

OpenLIT

OpenLIT is an open-source platform for AI engineering that provides LLM observability, prompt management, evaluations and guardrails.

Stars --Last Commit --

Opik

Opik is an open-source LLM evaluation and observability platform that helps teams build, evaluate and optimize LLM applications.

Stars --Last Commit --

Petri

Petri is an alignment auditing agent designed to quickly explore alignment hypotheses and help researchers automate evaluation workflows.

Stars --Last Commit --

Promptfoo

Promptfoo is a developer-first, local LLM testing and red-teaming tool for automated evaluations, vulnerability scanning, and CI integration.

Stars --Last Commit --

Ragas

Ragas is an open-source toolkit for evaluating and optimizing LLM applications, offering objective metrics, test data generation, and production feedback loops.

Stars --Last Commit --

ReLE Chinese LLM Benchmark

ReLE (chinese-llm-benchmark) is a continuously updated Chinese LLM evaluation and leaderboard project covering education, medical, finance, legal, reasoning and other capability dimensions.

Stars --Last Commit --

Shapash

Generates interactive visual reports to explain machine learning model predictions for stakeholders.

Stars --Last Commit --

YYLO Benchmark

YYLO Benchmark is the evaluation companion to the YYLO orchestrator — a trusted-host experiment runner for historical Ledger tasks, supplied coding prompts, and workflows. Each candidate attempt runs in a fresh repository with default exclusions for answer-bearing files, outputs are retained for later re-evaluation with new checks or judges, and frozen project/task checklists plus deterministic loss give reproducible comparison rows. It deliberately does not choose a winner or combine judge opinions with test results.

Stars --Last Commit --

Tracing, logging, and runtime observability.

Helicone

Helicone is an open-source LLM observability and analytics platform that captures requests, traces and sessions to help developers debug, evaluate and optimize model usage.

Stars --Last Commit --

Langfuse

Discover Langfuse, the open-source platform for LLM development, enhancing collaboration, monitoring, and debugging for AI applications.

Stars --Last Commit --

NeMo Relay

NVIDIA NeMo Relay is an open-source multi-language agent runtime and library for observing and controlling AI agent execution. It captures agent activity, applies middleware to model and tool calls, manages execution scopes and lifecycle events, and exports ATOF events, ATIF trajectories, or OpenTelemetry traces, logs, and metrics — without requiring changes to the existing agent stack.

Stars --Last Commit --

OpenLLMetry

An OpenTelemetry-inspired observability toolkit for LLM/AI, providing request tracing and metrics aggregation for diagnostics and monitoring.

Stars --Last Commit --

Phoenix

Phoenix is a high-performance web framework built with Elixir, optimized for realtime, distributed, and scalable web applications.

Stars --Last Commit --

Polyaxon

Polyaxon is an MLOps platform for managing, training and monitoring large-scale machine learning workloads.

Stars --Last Commit --

TraceRoot

TraceRoot is an open-source observability and self-improving layer for AI agents. It detects production failures, root-causes them against source code and GitHub history, opens verified fix PRs, and evals every fix.

Stars --Last Commit --

Prompt management, versioning, and quality tooling.

Code2Prompt

A tool that converts codebases into a single LLM prompt for code analysis, generation, and automation workflows.

Stars --Last Commit --

distilly

Distills a person's experience, judgment, voice, and ways of working into a portable, source-grounded Person Profile packaged as an Agent Skill that any compatible agent or bot can install and invoke. Formerly Colleague Skill.

Stars --Last Commit --

GuideLLM

GuideLLM offers tooling for guiding, interpreting, and controlling large language models (LLMs), enabling better controllability in interactive applications.

Stars --Last Commit --

PR-Agent

An open-source AI-powered code review and PR assistant that runs locally, in CI, or self-hosted; supports multi-platform integrations and customizable prompts.

Stars --Last Commit --

Safety filters, guardrails, and risk controls.

Anthropic Cybersecurity Skills

754 structured cybersecurity skills for AI agents mapped to 5 frameworks including MITRE ATT&CK, NIST CSF 2.0, and MITRE ATLAS.

Stars --Last Commit --

MCP Scanner

A tool to scan MCP servers and tools for potential security issues, using multi-engine analysis and customizable reporting.

Stars --Last Commit --

Security Audit Skill

A coding-agent skill from Cloudflare that orchestrates isolated sub-agents through a six-phase security audit pipeline, producing independently verified, machine-readable findings.

Stars --Last Commit --

SkillSpector

Security scanner for AI agent skills by NVIDIA that detects vulnerabilities, malicious patterns, and security risks across 64 vulnerability patterns in 16 categories.

Stars --Last Commit --

Compiler optimization, autotuning, and simulation.

Newton

A GPU-accelerated physics simulation engine built on NVIDIA Warp, targeting robotics and simulation research.

Stars --Last Commit --