#machine-learning

Powerful add-ons for PyMC

High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.

Time series forecasting with PyTorch

The AI Compute Platform for frontier teams. SkyPilot turns fragmented AI compute into one AI supercomputer, so frontier AI teams build custom intelligence faster.

Python Client for Supabase. Query Postgres from Flask, Django, FastAPI. Python user authentication, security policies, edge functions, file storage, and realtime data streaming. Good first issue.

GPU-capable global atmosphere model of the CliMA Earth System Model, designed for calibration with data assimilation and machine learning

Lightning fast data version control system for large repositories of data. Feels like git, pushes and pulls like oxen.

Python scraper based on AI

BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.

Library for reading and processing ML training data.

Visualise your Kedro data and machine-learning pipelines and track your experiments.

SDK libraries for Modal

Efficient in-memory representation for ONNX, in Python

Fast ML inference & training for ONNX models in Rust

A game theoretic approach to explain the output of any machine learning model.

Burn is a next generation tensor library and Deep Learning Framework that doesn't compromise on flexibility, efficiency and portability.

Unsloth is a local UI for training and running Kimi K3, Gemma 4, Qwen3.6, DeepSeek-V4, GLM and other models.

Streamlining reinforcement learning with RLOps. State-of-the-art RL algorithms and tools, with 10x faster training through evolutionary hyperparameter optimization.

Official front-end implementation of ComfyUI

⚡️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.

Chronos: Pretrained Models for Time Series Forecasting

ClearML - Auto-Magical CI/CD to streamline your AI workload. Experiment Management, Data Management, Pipeline, Orchestration, Scheduling & Serving in one MLOps/LLMOps solution

The Ersilia Model Hub, an open-source repository and CLI of AI/ML models for infectious and neglected disease research.

Fast, Flexible and Portable Structured Generation

Shapley Interactions and Shapley Values for Machine Learning

Agentic QE Fleet is an open-source AI-powered QA/QE platform designed for use with Coding Agents (works best with Claude Code) featuring specialized agents and skills to support testing activities for a product at any stage of the SDLC. Free to use, fork, build, and contribute. Based on the Agentic QE Framework created by Dragan Spiridonov.

DoWhy is a Python library for causal inference that supports explicit modeling and testing of causal assumptions. DoWhy is based on a unified language for causal inference, combining causal graphical models and potential outcomes frameworks.

Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

RuVector is a High Performance, Real-Time, Self-Learning Ai, Vector GNN, Memory DB built in Rust.