mila-iqia / milabench
Repository of machine learning benchmarks
☆23Updated this week
Related projects ⓘ
Alternatives and complementary repositories for milabench
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆60Updated 2 years ago
- Vectorization techniques for fast population-based training.☆54Updated 2 years ago
- Clockwork VAEs in JAX/Flax☆31Updated 3 years ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated last year
- flexible meta-learning in jax☆12Updated last year
- A simple hypernetwork implementation in jax using haiku.☆23Updated 2 years ago
- ☆20Updated 5 years ago
- GPT implementation in Flax☆18Updated 2 years ago
- Clean RL implementation using MLX☆26Updated 8 months ago
- Fast training of unitary deep network layers from low-rank updates☆28Updated last year
- An implementation of MuZero in JAX.☆53Updated 2 years ago
- A simple library for scaling up JAX programs☆125Updated last week
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 3 years ago
- PyTorch Package For Quasimetric Learning☆42Updated last week
- ☆34Updated 3 years ago
- Official codebase for Improving Computational Efficiency in Visual Reinforcement Learning via Stored Embeddings.☆21Updated 3 years ago
- Official repository for the paper "Zero-Shot AutoML with Pretrained Models"☆41Updated 10 months ago
- JAX implementations of core Deep RL algorithms☆79Updated 2 years ago
- Image augmentation library for Jax☆37Updated 7 months ago
- ☆17Updated 4 months ago
- ☆17Updated last year
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆30Updated 11 months ago
- Reinforcement Learning inside a 3D soccer simulation☆24Updated last month
- ☆26Updated 2 years ago
- Building blocks for productive research☆44Updated 2 weeks ago
- ☆48Updated 8 months ago
- A collection of meta-learning algorithms in Jax☆23Updated 2 years ago
- Automatically generate simple meta-learning tasks from a very large space☆15Updated last year
- WandB sweeps integration with Hydra sweeper☆46Updated 9 months ago
- ☆36Updated 2 years ago
- ☆62Updated 2 months ago