A ranked list of awesome machine learning Python libraries
An Easy-to-use, Scalable and High-performance RLHF Framework
A unified framework for scalable computing
[NeurIPS 2023 Spotlight] LightZero
A modular, primitive-first, python-first PyTorch library
PyTorch version of Stable Baselines
A Modular Simulation Framework and Benchmark for Robot Learning
Open-source, high-performance AI model with advanced reasoning
TextWorld is a sandbox learning environment for the training
An API standard for single-agent reinforcement learning environments
A TensorFlow library for applied reinforcement learning
agentUniverse is a LLM multi-agent framework
Flexible and powerful framework for managing multiple AI agents
A modular high-level library to train embodied AI agents
Volcano Engine Reinforcement Learning for LLMs
MedicalGPT: Training Your Own Medical GPT Model with ChatGPT Training
Simulation of spiking neural networks (SNNs) using PyTorch
Powerful AI language model (MoE) optimized for efficiency/performance
Monte Carlo tree search in JAX
Framework and no-code GUI for fine-tuning LLMs
Training framework for Stable Baselines3 reinforcement learning agents
Jittor is a high-performance deep learning framework
Tool for visualizing and tracking your machine learning experiments
Massively parallel rigidbody physics simulation
An Open Source package that allows video game creators