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
Massively parallel rigidbody physics simulation
Machine learning system which pushes the frontier of machine learning
OpenDILab Decision AI Engine
Algorithms for explaining machine learning models
Tool for visualizing and tracking your machine learning experiments
Agent S: an open agentic framework that uses computers like a human
RL implementations
An API standard for single-agent reinforcement learning environments
Cosmos-RL is a flexible and scalable Reinforcement Learning framework
Implementation of RLHF (Reinforcement Learning with Human Feedback)
PyTorch version of Stable Baselines
TextWorld is a sandbox learning environment for the training
agentUniverse is a LLM multi-agent framework
Simulation of spiking neural networks (SNNs) using PyTorch
VMAS is a vectorized differentiable simulator
Flexible and powerful framework for managing multiple AI agents
MuA multi-agent reinforcement learning environment
Modular Deep Reinforcement Learning framework in PyTorch
MedicalGPT: Training Your Own Medical GPT Model with ChatGPT Training
Framework and no-code GUI for fine-tuning LLMs