A ranked list of awesome machine learning Python libraries
A modular, primitive-first, python-first PyTorch library
agentUniverse is a LLM multi-agent framework
Simulation of spiking neural networks (SNNs) using PyTorch
A unified framework for scalable computing
Flexible and powerful framework for managing multiple AI agents
An API standard for single-agent reinforcement learning environments
VMAS is a vectorized differentiable simulator
A modular high-level library to train embodied AI agents
A TensorFlow library for applied reinforcement learning
An Open Source package that allows video game creators
Jittor is a high-performance deep learning framework
PyTorch version of Stable Baselines
[NeurIPS 2023 Spotlight] LightZero
Massively parallel rigidbody physics simulation
OpenDILab Decision AI Engine
An Easy-to-use, Scalable and High-performance RLHF Framework
RL implementations
DeepMind's software stack for physics-based simulation
RL research on Android devices
Monte Carlo tree search in JAX
A Modular Simulation Framework and Benchmark for Robot Learning
A minimalist environment for decision-making in autonomous driving
Powerful AI language model (MoE) optimized for efficiency/performance
Training framework for Stable Baselines3 reinforcement learning agents