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
[NeurIPS 2023 Spotlight] LightZero
Algorithms for explaining machine learning models
Massively parallel rigidbody physics simulation
Agent S: an open agentic framework that uses computers like a human
MedicalGPT: Training Your Own Medical GPT Model with ChatGPT Training
DeepMind's software stack for physics-based simulation
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
Framework and no-code GUI for fine-tuning LLMs
RL implementations
TextWorld is a sandbox learning environment for the training
Implementation of RLHF (Reinforcement Learning with Human Feedback)
SAPIEN Manipulation Skill Framework
Simulation of spiking neural networks (SNNs) using PyTorch
Flexible and powerful framework for managing multiple AI agents
Cosmos-RL is a flexible and scalable Reinforcement Learning framework
PyTorch version of Stable Baselines
Training framework for Stable Baselines3 reinforcement learning agents
Modular Deep Reinforcement Learning framework in PyTorch
Language Model Reinforcement Learning Environments frameworks
A Modular Simulation Framework and Benchmark for Robot Learning