A ranked list of awesome machine learning Python libraries
A TensorFlow library for applied reinforcement learning
Simulation of spiking neural networks (SNNs) using PyTorch
A modular, primitive-first, python-first PyTorch library
A unified framework for scalable computing
An Open Source package that allows video game creators
Jittor is a high-performance deep learning framework
Massively parallel rigidbody physics simulation
[NeurIPS 2023 Spotlight] LightZero
An Easy-to-use, Scalable and High-performance RLHF Framework
TextWorld is a sandbox learning environment for the training
DeepMind's software stack for physics-based simulation
An API standard for single-agent reinforcement learning environments
Monte Carlo tree search in JAX
An API standard for multi-agent reinforcement learning environments
RL research on Android devices
A Modular Simulation Framework and Benchmark for Robot Learning
PyTorch version of Stable Baselines
agentUniverse is a LLM multi-agent framework
Flexible and powerful framework for managing multiple AI agents
Advanced evolutionary computation library built on top of PyTorch
SAPIEN Manipulation Skill Framework
Training framework for Stable Baselines3 reinforcement learning agents
MuA multi-agent reinforcement learning environment
RL implementations