#machine-learning

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

Official front-end implementation of ComfyUI

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

Cross-platform, customizable ML solutions for live and streaming media.

Efficient in-memory representation for ONNX, in Python

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

Run, manage, and scale AI workloads on any AI infrastructure. Use one system to access & manage all AI compute (Kubernetes, Slurm, 20+ clouds, on-prem).

Probabilistic time series modeling in Python

A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.

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

Limen unifies parameter search across machine learning and rule-based strategies, with built-in analytics that show not just what works, but why it works.

Fast and accurate AI powered file content types detection

skops is a Python library helping you share your scikit-learn based models and put them in production

Stanford NLP Python library for tokenization, sentence segmentation, NER, and parsing of many human languages

An Open Source Machine Learning Framework for Everyone

Chronos: Pretrained Models for Time Series Forecasting

MQT Predictor - A Tool for Automatic Device Selection with Device-Specific Circuit Compilation for Quantum Computing

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

Relax! Flux is the ML library that doesn't make you tensor

Tensor library for machine learning

Data manipulation and transformation for audio signal processing, powered by PyTorch

Reservoir computing utilities for scientific machine learning (SciML)

A fully open-source humanoid arm for physical AI research and deployment in contact-rich environments.

JSON schemas and examples representing structural data, characteristic properties, modeling workflows and related data about materials standardizing the diverse landscape of information

TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and support for E2E production ML pipelines when you're ready.

A Python Package to Tackle the Curse of Imbalanced Datasets in Machine Learning

TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...

Flexible and powerful framework for managing multiple AI agents and handling complex conversations

High-performance automatic differentiation of LLVM and MLIR.

The fastai deep learning library