teddykoker / tinyloader
☆63Updated 2 years ago
Alternatives and similar repositories for tinyloader:
Users that are interested in tinyloader are comparing it to the libraries listed below
- JAX implementation of Learning to learn by gradient descent by gradient descent☆27Updated 4 months ago
- A collection of optimizers, some arcane others well known, for Flax.☆29Updated 3 years ago
- Implementation of GateLoop Transformer in Pytorch and Jax☆87Updated 8 months ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆83Updated last year
- ☆30Updated 3 months ago
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆30Updated 2 months ago
- A place to store reusable transformer components of my own creation or found on the interwebs☆47Updated 2 weeks ago
- Machine Learning eXperiment Utilities☆46Updated 9 months ago
- Distributed ML Optimizer☆30Updated 3 years ago
- Implementation of the Kalman Filtering Attention proposed in "Kalman Filtering Attention for User Behavior Modeling in CTR Prediction"☆57Updated last year
- code for the ddp tutorial☆32Updated 2 years ago
- Simply Numpy implementation of the FAVOR+ attention mechanism, https://teddykoker.com/2020/11/performers/☆37Updated 4 years ago
- A case study of efficient training of large language models using commodity hardware.☆68Updated 2 years ago
- Implémentation of the article **Deep Learning CUDA Memory Usage and Pytorch optimization tricks**☆43Updated 5 years ago
- A collection of Models, Datasets, DataModules, Callbacks, Metrics, Losses and Loggers to better integrate pytorch-lightning with transfor…☆47Updated last year
- Blog post☆17Updated last year
- An implementation of Transformer with Expire-Span, a circuit for learning which memories to retain☆33Updated 4 years ago
- Dense Passage Retrieval using tensorflow-keras on TPU☆15Updated 3 years ago
- Implementation of Token Shift GPT - An autoregressive model that solely relies on shifting the sequence space for mixing☆48Updated 3 years ago
- PyTorch implementation of GLOM☆21Updated 2 years ago
- This repository contains example code to build models on TPUs☆30Updated 2 years ago
- a lightweight transformer library for PyTorch☆71Updated 3 years ago
- Large Scale Distributed Model Training strategy with Colossal AI and Lightning AI☆57Updated last year
- ☆73Updated 2 years ago
- Fast Discounted Cumulative Sums in PyTorch☆95Updated 3 years ago
- My explorations into editing the knowledge and memories of an attention network☆34Updated 2 years ago
- Make triton easier☆47Updated 9 months ago
- AdaCat☆49Updated 2 years ago
- ☆20Updated last year