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.☆283Updated last year
- a Lightweight library for sequential learning agents, including reinforcement learning☆426Updated last year
- PIX is an image processing library in JAX, for JAX.☆397Updated last month
- Named tensors with first-class dimensions for PyTorch☆322Updated last year
- ☆317Updated last year
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes☆237Updated last year
- RLStructures is a library to facilitate the implementation of new reinforcement learning algorithms. It includes a library, a tutorial, a…☆258Updated last year
- Pytorch Lightning Distributed Accelerators using Ray☆211Updated last year
- ☆340Updated this week
- A High Level API for Deep Learning in JAX☆471Updated 2 years ago
- ☆806Updated this week
- ☆153Updated last year
- ☆100Updated 11 months ago
- CLU lets you write beautiful training loops in JAX.☆329Updated this week
- ☆335Updated 9 months ago
- A library to inspect and extract intermediate layers of PyTorch models.☆470Updated 2 years ago
- A platform for managing machine learning experiments☆832Updated last month
- functorch is JAX-like composable function transforms for PyTorch.☆1,403Updated this week
- ☆545Updated last month
- [Prototype] Tools for the concurrent manipulation of variably sized Tensors.☆253Updated 2 years ago
- Train very large language models in Jax.☆198Updated last year
- A Pytree Module system for Deep Learning in JAX☆213Updated last year
- 🎢 Creating and sharing simulation environments for embodied and synthetic data research☆190Updated last year
- Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.☆240Updated last year
- TensorDict is a pytorch dedicated tensor container.☆862Updated this week
- jax-triton contains integrations between JAX and OpenAI Triton☆363Updated this week
- Lightweight Hyperparameter Optimization 🚂☆143Updated 4 months ago
- Extremely Fast End-to-End Deep Multi-Agent Reinforcement Learning Framework on a GPU (JMLR 2022)☆469Updated 5 months ago
- ☆153Updated 4 years ago
- This project was moved to: https://github.com/coax-dev/coax☆160Updated 2 years ago