facebookresearch / MathsFromExamples
Source code, datasets and trained models for the paper Learning Advanced Mathematical Computations from Examples (ICLR 2021), by François Charton, Amaury Hayat (ENPC-Rutgers) and Guillaume Lample
☆178Updated 3 years ago
Related projects ⓘ
Alternatives and complementary repositories for MathsFromExamples
- ☆155Updated 4 years ago
- RLStructures is a library to facilitate the implementation of new reinforcement learning algorithms. It includes a library, a tutorial, a…☆259Updated last year
- A Pytree Module system for Deep Learning in JAX☆214Updated last year
- ☆153Updated 11 months ago
- PIX is an image processing library in JAX, for JAX.☆389Updated last week
- This project was moved to: https://github.com/coax-dev/coax☆160Updated last year
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆175Updated 2 years ago
- Docs☆143Updated last month
- A library for distributed ML training with PyTorch☆366Updated last year
- ☆67Updated last year
- Fast Block Sparse Matrices for Pytorch☆545Updated 3 years ago
- a Lightweight library for sequential learning agents, including reinforcement learning☆427Updated last year
- ☆101Updated 3 years ago
- ☆536Updated 2 months ago
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes☆237Updated last year
- Course webpage for COMP 790, (Deep) Learning from Limited Labeled Data☆302Updated 4 years ago
- Repository for tutorial sessions at EEML2020☆271Updated 3 years ago
- ☆191Updated 6 months ago
- 🧀 Pytorch code for the Fromage optimiser.☆122Updated 4 months ago
- A High Level API for Deep Learning in JAX☆470Updated last year
- ☆788Updated this week
- A PyTorch reinforcement learning library for generalizable and reproducible algorithm implementations with an aim to improve accessibilit…☆404Updated last year
- ☆105Updated 2 weeks ago
- a lightweight transformer library for PyTorch☆72Updated 3 years ago
- JMP is a Mixed Precision library for JAX.☆187Updated 6 months ago
- Lightweight Hyperparameter Optimization 🚂☆145Updated 2 months ago
- Mixture Density Networks (Bishop, 1994) tutorial in JAX☆58Updated 4 years ago
- Example python package with pybind11 cpp extension☆57Updated 3 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆104Updated 2 years ago
- Named tensors with first-class dimensions for PyTorch☆322Updated last year