Powerful AI language model (MoE) optimized for efficiency/performance
Open-source, high-performance AI model with advanced reasoning
DeepMind's software stack for physics-based simulation
Simulation of spiking neural networks (SNNs) using PyTorch
VMAS is a vectorized differentiable simulator
Agent S: an open agentic framework that uses computers like a human
An API standard for single-agent reinforcement learning environments
Framework and no-code GUI for fine-tuning LLMs
TextWorld is a sandbox learning environment for the training
Benchmarking Multimodal Agents for Open-Ended Tasks
A unified framework for scalable computing
Monte Carlo tree search in JAX
RL implementations
OpenDILab Decision AI Engine
A minimalist environment for decision-making in autonomous driving
A Modular Simulation Framework and Benchmark for Robot Learning
An API standard for multi-agent reinforcement learning environments
Training framework for Stable Baselines3 reinforcement learning agents
MuA multi-agent reinforcement learning environment
A TensorFlow library for applied reinforcement learning
[NeurIPS 2023 Spotlight] LightZero
Modular Deep Reinforcement Learning framework in PyTorch
Advanced evolutionary computation library built on top of PyTorch
Implementation of RLHF (Reinforcement Learning with Human Feedback)
Massively parallel rigidbody physics simulation