A ranked list of awesome machine learning Python libraries
A unified framework for scalable computing
Jittor is a high-performance deep learning framework
Modular Deep Reinforcement Learning framework in PyTorch
A TensorFlow library for applied reinforcement learning
Monte Carlo tree search in JAX
Physical Symbolic Optimization
A modular, primitive-first, python-first PyTorch library
Algorithms for explaining machine learning models
An Easy-to-use, Scalable and High-performance RLHF Framework
DeepMind's software stack for physics-based simulation
Simulation of spiking neural networks (SNNs) using PyTorch
[NeurIPS 2023 Spotlight] LightZero
agentUniverse is a LLM multi-agent framework
Massively parallel rigidbody physics simulation
TextWorld is a sandbox learning environment for the training
A Modular Simulation Framework and Benchmark for Robot Learning
An Open Source package that allows video game creators
PyTorch version of Stable Baselines
An API standard for single-agent reinforcement learning environments
Tool for visualizing and tracking your machine learning experiments
RL research on Android devices
Flexible and powerful framework for managing multiple AI agents
MuA multi-agent reinforcement learning environment
Benchmarking Multimodal Agents for Open-Ended Tasks