Monte Carlo tree search in JAX
[NeurIPS 2023 Spotlight] LightZero
Advanced evolutionary computation library built on top of PyTorch
Massively parallel rigidbody physics simulation
MedicalGPT: Training Your Own Medical GPT Model with ChatGPT Training
Volcano Engine Reinforcement Learning for LLMs
DeepMind's software stack for physics-based simulation
Modular Deep Reinforcement Learning framework in PyTorch
SAPIEN Manipulation Skill Framework
OpenDILab Decision AI Engine
Training framework for Stable Baselines3 reinforcement learning agents
Implementation of RLHF (Reinforcement Learning with Human Feedback)
Cosmos-RL is a flexible and scalable Reinforcement Learning framework
Language Model Reinforcement Learning Environments frameworks
MuA multi-agent reinforcement learning environment
Physical Symbolic Optimization
Benchmarking Multimodal Agents for Open-Ended Tasks
Optical-packet node transceiver frequency allocation
Algorithms for explaining machine learning models
A code-first agent framework for seamlessly planning analytics tasks
A collection of reference Jupyter notebooks and demo AI/ML application
The most simple, flexible, and comprehensive OpenAI Gym trading
Reinforcement learning (RL) tutorial series
A repo for distributed training of language models with Reinforcement
High-quality single-file implementations of SOTA Offline