vinayprabhu / X-is-all-you-need
A survey of all the 'X is all you need'
☆72Updated 3 years ago
Alternatives and similar repositories for X-is-all-you-need:
Users that are interested in X-is-all-you-need are comparing it to the libraries listed below
- a lightweight transformer library for PyTorch☆71Updated 3 years ago
- A GPT, made only of MLPs, in Jax☆57Updated 3 years ago
- A case study of efficient training of large language models using commodity hardware.☆68Updated 2 years ago
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆84Updated last year
- Differentiable Algorithms and Algorithmic Supervision.☆112Updated last year
- PyTorch centric eager mode debugger☆46Updated 2 months ago
- 👑 Pytorch code for the Nero optimiser.☆20Updated 2 years ago
- A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-…☆66Updated last year
- Official repository for the paper "Going Beyond Linear Transformers with Recurrent Fast Weight Programmers" (NeurIPS 2021)☆48Updated last year
- Image augmentation library for Jax☆37Updated 10 months ago
- Large dataset storage format for Pytorch☆45Updated 3 years ago
- A collection of optimizers, some arcane others well known, for Flax.☆29Updated 3 years ago
- Lightweight Cluster/Cloud VM Job Management 🚀☆39Updated 5 months ago
- ☆67Updated last year
- Torch Distributed Experimental☆115Updated 6 months ago
- Supplementary material for our paper "Compute Trends Across Three Eras of Machine Learning".☆38Updated 2 years ago
- Annotate python source code☆69Updated 4 years ago
- Code for the anonymous submission "Cockpit: A Practical Debugging Tool for Training Deep Neural Networks"☆31Updated 4 years ago
- A place to store reusable transformer components of my own creation or found on the interwebs☆44Updated this week
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 3 years ago
- Lightweight ML Experiment Logging 📖☆78Updated 5 months ago
- pytest plugin for a better developer experience when working with the PyTorch test suite☆44Updated 3 years ago
- A simple Transformer where the softmax has been replaced with normalization☆19Updated 4 years ago
- My implementation of DeepMind's Perceiver☆61Updated 3 years ago
- Distributed ML Optimizer☆30Updated 3 years ago
- Named tensors with first-class dimensions for PyTorch☆321Updated last year
- A framework for implementing equivariant DL☆10Updated 3 years ago
- ☆39Updated 2 years ago
- Convert scikit-learn models to PyTorch modules☆160Updated 9 months ago
- 👩 Pytorch and Jax code for the Madam optimiser.☆51Updated 4 years ago