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
[NeurIPS 2023 Spotlight] LightZero
Tool for visualizing and tracking your machine learning experiments
Massively parallel rigidbody physics simulation
RL implementations
Algorithms for explaining machine learning models
OpenDILab Decision AI Engine
TextWorld is a sandbox learning environment for the training
Implementation of RLHF (Reinforcement Learning with Human Feedback)
An Easy-to-use, Scalable and High-performance RLHF Framework
Agent S: an open agentic framework that uses computers like a human
Simulation of spiking neural networks (SNNs) using PyTorch
DeepMind's software stack for physics-based simulation
agentUniverse is a LLM multi-agent framework
MedicalGPT: Training Your Own Medical GPT Model with ChatGPT Training
Cosmos-RL is a flexible and scalable Reinforcement Learning framework
VMAS is a vectorized differentiable simulator
MuA multi-agent reinforcement learning environment
PyTorch version of Stable Baselines
Modular Deep Reinforcement Learning framework in PyTorch
SAPIEN Manipulation Skill Framework
An API standard for single-agent reinforcement learning environments
Flexible and powerful framework for managing multiple AI agents