facebookresearch / moolib
A library for distributed ML training with PyTorch
☆366Updated 2 years ago
Alternatives and similar repositories for moolib:
Users that are interested in moolib are comparing it to the libraries listed below
- RLMeta is a light-weight flexible framework for Distributed Reinforcement Learning Research.☆282Updated 2 years ago
- a Lightweight library for sequential learning agents, including reinforcement learning☆429Updated last year
- PIX is an image processing library in JAX, for JAX.☆411Updated 2 weeks ago
- RLStructures is a library to facilitate the implementation of new reinforcement learning algorithms. It includes a library, a tutorial, a…☆260Updated 2 years ago
- Named tensors with first-class dimensions for PyTorch☆321Updated last year
- ☆319Updated last year
- ☆100Updated last year
- A platform for managing machine learning experiments☆840Updated this week
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes☆236Updated last year
- Reinforcement learning library in JAX.☆100Updated last year
- ☆555Updated 3 weeks ago
- Pytorch Lightning Distributed Accelerators using Ray☆211Updated last year
- ☆154Updated last year
- A High Level API for Deep Learning in JAX☆472Updated 2 years ago
- ☆345Updated this week
- A library to inspect and extract intermediate layers of PyTorch models.☆472Updated 2 years ago
- Extremely Fast End-to-End Deep Multi-Agent Reinforcement Learning Framework on a GPU (JMLR 2022)☆478Updated 7 months ago
- functorch is JAX-like composable function transforms for PyTorch.☆1,414Updated this week
- ☆1,296Updated last month
- CLU lets you write beautiful training loops in JAX.☆335Updated 2 weeks ago
- A Pytree Module system for Deep Learning in JAX☆213Updated 2 years ago
- Fast Block Sparse Matrices for Pytorch☆546Updated 4 years ago
- Cockpit: A Practical Debugging Tool for Training Deep Neural Networks☆474Updated 2 years ago
- 🎢 Creating and sharing simulation environments for embodied and synthetic data research☆191Updated last year
- ☆832Updated this week
- ☆153Updated 4 years ago
- Rax is a Learning-to-Rank library written in JAX.☆327Updated 2 weeks ago
- Modular framework for Reinforcement Learning in python☆172Updated 2 years ago
- Library for 8-bit optimizers and quantization routines.☆718Updated 2 years ago
- Unofficial JAX implementations of deep learning research papers☆153Updated 2 years ago