A ranked list of awesome machine learning Python libraries
bsuite is a collection of carefully-designed experiments
A modular, primitive-first, python-first PyTorch library
Train multi-step agents for real-world tasks using GRPO
Algorithms for explaining machine learning models
Simulation of spiking neural networks (SNNs) using PyTorch
An API standard for single-agent reinforcement learning environments
A unified framework for scalable computing
A TensorFlow library for applied reinforcement learning
[NeurIPS 2023 Spotlight] LightZero
An Easy-to-use, Scalable and High-performance RLHF Framework
Massively parallel rigidbody physics simulation
An Open Source package that allows video game creators
MuA multi-agent reinforcement learning environment
agentUniverse is a LLM multi-agent framework
Flexible and powerful framework for managing multiple AI agents
Jittor is a high-performance deep learning framework
DeepMind's software stack for physics-based simulation
PyTorch version of Stable Baselines
A code-first agent framework for seamlessly planning analytics tasks
Physical Symbolic Optimization
An API standard for multi-agent reinforcement learning environments
Monte Carlo tree search in JAX
MedicalGPT: Training Your Own Medical GPT Model with ChatGPT Training
A modular high-level library to train embodied AI agents