A ranked list of awesome machine learning Python libraries
PyTorch version of Stable Baselines
agentUniverse is a LLM multi-agent framework
A modular, primitive-first, python-first PyTorch library
Algorithms for explaining machine learning models
Flexible and powerful framework for managing multiple AI agents
An Easy-to-use, Scalable and High-performance RLHF Framework
MuA multi-agent reinforcement learning environment
Simulation of spiking neural networks (SNNs) using PyTorch
Powerful AI language model (MoE) optimized for efficiency/performance
A modular high-level library to train embodied AI agents
VMAS is a vectorized differentiable simulator
A unified framework for scalable computing
[NeurIPS 2023 Spotlight] LightZero
MedicalGPT: Training Your Own Medical GPT Model with ChatGPT Training
A TensorFlow library for applied reinforcement learning
Massively parallel rigidbody physics simulation
Framework and no-code GUI for fine-tuning LLMs
DeepMind's software stack for physics-based simulation
An API standard for single-agent reinforcement learning environments
Open-source, high-performance AI model with advanced reasoning
A code-first agent framework for seamlessly planning analytics tasks
Jittor is a high-performance deep learning framework
A minimalist environment for decision-making in autonomous driving
Physical Symbolic Optimization