teddykoker / tinyloaderLinks
☆66Updated 5 months ago
Alternatives and similar repositories for tinyloader
Users that are interested in tinyloader are comparing it to the libraries listed below
Sorting:
- A case study of efficient training of large language models using commodity hardware.☆68Updated 3 years ago
- HomebrewNLP in JAX flavour for maintable TPU-Training☆50Updated last year
- Amos optimizer with JEstimator lib.☆82Updated last year
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆87Updated last year
- Implementation of VQ-VAE with a GPT-style sampler in the JAX and Haiku ecosystem.☆12Updated last year
- ☆36Updated 3 years ago
- Fast Discounted Cumulative Sums in PyTorch☆96Updated 4 years ago
- Implementation of GateLoop Transformer in Pytorch and Jax☆90Updated last year
- JAX implementation of Learning to learn by gradient descent by gradient descent☆27Updated 3 weeks ago
- ML/DL Math and Method notes☆63Updated last year
- Functional deep learning☆108Updated 2 years ago
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆83Updated last year
- some common Huggingface transformers in maximal update parametrization (µP)☆82Updated 3 years ago
- Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)☆187Updated 3 years ago
- A GPT, made only of MLPs, in Jax☆58Updated 4 years ago
- A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-…☆67Updated 2 years ago
- A deep learning library based on Pytorch focussed on low resource language research and robustness☆70Updated 3 years ago
- Torch Distributed Experimental☆117Updated last year
- Implementation of TableFormer, Robust Transformer Modeling for Table-Text Encoding, in Pytorch☆39Updated 3 years ago
- Large Scale Distributed Model Training strategy with Colossal AI and Lightning AI☆56Updated 2 years ago
- Implementation of Flash Attention in Jax☆216Updated last year
- JAX/Flax implementation of the Hyena Hierarchy☆34Updated 2 years ago
- A collection of optimizers, some arcane others well known, for Flax.☆29Updated 4 years ago
- Official repository for the paper "Zero-Shot AutoML with Pretrained Models"☆47Updated last year
- Machine Learning eXperiment Utilities☆46Updated last month
- ☆31Updated 2 months ago
- This repository contains example code to build models on TPUs☆30Updated 2 years ago
- A place to store reusable transformer components of my own creation or found on the interwebs☆60Updated last week
- Convert scikit-learn models to PyTorch modules☆164Updated last year
- An implementation of Transformer with Expire-Span, a circuit for learning which memories to retain☆34Updated 4 years ago