teddykoker / tinyloaderLinks
☆67Updated 7 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
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆82Updated 2 years ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆89Updated last year
- A deep learning library based on Pytorch focussed on low resource language research and robustness☆70Updated 3 years ago
- Implementation of the Kalman Filtering Attention proposed in "Kalman Filtering Attention for User Behavior Modeling in CTR Prediction"☆59Updated 2 years ago
- Simply Numpy implementation of the FAVOR+ attention mechanism, https://teddykoker.com/2020/11/performers/☆38Updated 4 years ago
- ☆31Updated last week
- Torch Distributed Experimental☆117Updated last year
- A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-…☆66Updated 2 years ago
- My explorations into editing the knowledge and memories of an attention network☆35Updated 2 years ago
- a lightweight transformer library for PyTorch☆72Updated 4 years ago
- Functional deep learning☆108Updated 3 years ago
- ☆36Updated 3 years ago
- Amos optimizer with JEstimator lib.☆82Updated last year
- ML/DL Math and Method notes☆64Updated last year
- A place to store reusable transformer components of my own creation or found on the interwebs☆62Updated last month
- A library to create and manage configuration files, especially for machine learning projects.☆80Updated 3 years ago
- Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)☆188Updated 3 years ago
- An implementation of Transformer with Expire-Span, a circuit for learning which memories to retain☆34Updated 5 years ago
- HomebrewNLP in JAX flavour for maintable TPU-Training☆51Updated last year
- Fast Discounted Cumulative Sums in PyTorch☆96Updated 4 years ago
- Implementation of GateLoop Transformer in Pytorch and Jax☆90Updated last year
- A collection of optimizers, some arcane others well known, for Flax.☆29Updated 4 years ago
- Implementation of VQ-VAE with a GPT-style sampler in the JAX and Haiku ecosystem.☆12Updated 2 years ago
- JAX/Flax implementation of the Hyena Hierarchy☆34Updated 2 years ago
- some common Huggingface transformers in maximal update parametrization (µP)☆86Updated 3 years ago
- Differentiable Algorithms and Algorithmic Supervision.☆116Updated 2 years ago
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆32Updated 5 months ago
- Implementation of Token Shift GPT - An autoregressive model that solely relies on shifting the sequence space for mixing☆50Updated 3 years ago
- Implementation of TableFormer, Robust Transformer Modeling for Table-Text Encoding, in Pytorch☆39Updated 3 years ago