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
Algorithms for explaining machine learning models
Simulation of spiking neural networks (SNNs) using PyTorch
Train multi-step agents for real-world tasks using GRPO
A TensorFlow library for applied reinforcement learning
An API standard for single-agent reinforcement learning environments
A unified framework for scalable computing
An Easy-to-use, Scalable and High-performance RLHF Framework
MuA multi-agent reinforcement learning environment
[NeurIPS 2023 Spotlight] LightZero
An Open Source package that allows video game creators
Massively parallel rigidbody physics simulation
Flexible and powerful framework for managing multiple AI agents
PyTorch version of Stable Baselines
agentUniverse is a LLM multi-agent framework
Jittor is a high-performance deep learning framework
DeepMind's software stack for physics-based simulation
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
Environments and algorithms for research in general reinforcement
RL research on Android devices