Fast, flexible and easy to use probabilistic modelling in Python
Hummingbird compiles trained ML models into tensor computation
A very simple framework for state-of-the-art NLP
Label, clean and enrich text datasets with LLMs
Open source libraries and APIs to build custom preprocessing pipelines
Semantic cache for LLMs. Fully integrated with LangChain
Open Source Differentiable Computer Vision Library
TensorFlow-based neural network library
A fast library for AutoML and tuning
Decentralized deep learning in PyTorch. Built to train models
A unified, comprehensive and efficient recommendation library
Image processing in Python
The unified and scalable ML library for large-scale training
A machine learning library for detecting anomalies in signals
A library for scientific machine learning & physics-informed learning
PyTorch extensions for fast R&D prototyping and Kaggle farming
Probabilistic reasoning and statistical analysis in TensorFlow
Medical imaging toolkit for deep learning
DeepMind's software stack for physics-based simulation
The largest collection of PyTorch image encoders / backbones
Advanced evolutionary computation library built on top of PyTorch
An API standard for multi-agent reinforcement learning environments
Training framework for Stable Baselines3 reinforcement learning agents
Monte Carlo tree search in JAX
The Library for LLM-based multi-agent applications