Powerful AI language model (MoE) optimized for efficiency/performance
Open-source, high-performance AI model with advanced reasoning
A modular, primitive-first, python-first PyTorch library
C++-based high-performance parallel environment execution engine
[NeurIPS 2023 Spotlight] LightZero
TextWorld is a sandbox learning environment for the training
Massively parallel rigidbody physics simulation
Algorithms for explaining machine learning models
OpenDILab Decision AI Engine
An API standard for single-agent reinforcement learning environments
Agent S: an open agentic framework that uses computers like a human
MuA multi-agent reinforcement learning environment
RL implementations
Cosmos-RL is a flexible and scalable Reinforcement Learning framework
Implementation of RLHF (Reinforcement Learning with Human Feedback)
agentUniverse is a LLM multi-agent framework
Flexible and powerful framework for managing multiple AI agents
Simulation of spiking neural networks (SNNs) using PyTorch
Tool for visualizing and tracking your machine learning experiments
VMAS is a vectorized differentiable simulator
PyTorch version of Stable Baselines
Machine learning system which pushes the frontier of machine learning
Modular Deep Reinforcement Learning framework in PyTorch
Language Model Reinforcement Learning Environments frameworks
SAPIEN Manipulation Skill Framework