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
Agent S: an open agentic framework that uses computers like a human
MuA multi-agent reinforcement learning environment
OpenDILab Decision AI Engine
TextWorld is a sandbox learning environment for the training
RL implementations
An API standard for single-agent reinforcement learning environments
Massively parallel rigidbody physics simulation
Simulation of spiking neural networks (SNNs) using PyTorch
Cosmos-RL is a flexible and scalable Reinforcement Learning framework
Implementation of RLHF (Reinforcement Learning with Human Feedback)
PyTorch version of Stable Baselines
Training framework for Stable Baselines3 reinforcement learning agents
Flexible and powerful framework for managing multiple AI agents
Modular Deep Reinforcement Learning framework in PyTorch
VMAS is a vectorized differentiable simulator
Algorithms for explaining machine learning models
agentUniverse is a LLM multi-agent framework
Tool for visualizing and tracking your machine learning experiments
Language Model Reinforcement Learning Environments frameworks
DeepMind's software stack for physics-based simulation
A modular high-level library to train embodied AI agents