mila-iqia / milabench
Repository of machine learning benchmarks
☆32Updated this week
Related projects ⓘ
Alternatives and complementary repositories for milabench
- Scaling scaling laws with board games.☆43Updated last year
- A simple library for scaling up JAX programs☆127Updated 2 weeks ago
- ☆29Updated 2 months ago
- ☆48Updated 9 months ago
- JAX Arrays for human consumption☆88Updated last year
- ☆105Updated 2 weeks ago
- Proof-of-concept of global switching between numpy/jax/pytorch in a library.☆18Updated 5 months ago
- Code for "Meta Learning Backpropagation And Improving It" @ NeurIPS 2021 https://arxiv.org/abs/2012.14905☆31Updated 2 years ago
- Running Jax in PyTorch Lightning☆82Updated 2 weeks ago
- Accelerated replay buffers in JAX☆40Updated 2 years ago
- Code for minimum-entropy coupling.☆30Updated 4 months ago
- Personal solutions to the Triton Puzzles☆16Updated 4 months ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- ☆36Updated 10 months ago
- ☆17Updated last year
- Building blocks for productive research☆45Updated last week
- General Modules for JAX☆58Updated 3 months ago
- If it quacks like a tensor...☆52Updated last week
- ☆17Updated 5 months ago
- ☆18Updated 7 months ago
- Generative cellular automaton-like learning environments for RL.☆19Updated last month
- Einsum-like high-level array sharding API for JAX☆32Updated 4 months ago
- Experiment of using Tangent to autodiff triton☆72Updated 10 months ago
- flexible meta-learning in jax☆12Updated last year
- Image augmentation library for Jax☆37Updated 7 months ago
- A collection of meta-learning algorithms in Jax☆23Updated 2 years ago
- ☆36Updated 2 years ago
- Minimal but scalable implementation of large language models in JAX☆26Updated 2 weeks ago