Powerful AI language model (MoE) optimized for efficiency/performance
A modular, primitive-first, python-first PyTorch library
Open-source, high-performance AI model with advanced reasoning
C++-based high-performance parallel environment execution engine
[NeurIPS 2023 Spotlight] LightZero
Massively parallel rigidbody physics simulation
DeepMind's software stack for physics-based simulation
Algorithms for explaining machine learning models
agentUniverse is a LLM multi-agent framework
Tool for visualizing and tracking your machine learning experiments
An Easy-to-use, Scalable and High-performance RLHF Framework
Machine learning system which pushes the frontier of machine learning
MedicalGPT: Training Your Own Medical GPT Model with ChatGPT Training
Agent S: an open agentic framework that uses computers like a human
TextWorld is a sandbox learning environment for the training
Framework and no-code GUI for fine-tuning LLMs
RL implementations
PyTorch version of Stable Baselines
Implementation of RLHF (Reinforcement Learning with Human Feedback)
SAPIEN Manipulation Skill Framework
Flexible and powerful framework for managing multiple AI agents
A Modular Simulation Framework and Benchmark for Robot Learning
Benchmarking Multimodal Agents for Open-Ended Tasks
Simulation of spiking neural networks (SNNs) using PyTorch
VMAS is a vectorized differentiable simulator