teddykoker / tinyloaderLinks
☆67Updated 6 months ago
Alternatives and similar repositories for tinyloader
Users that are interested in tinyloader are comparing it to the libraries listed below
Sorting:
- Implementation of VQ-VAE with a GPT-style sampler in the JAX and Haiku ecosystem.☆12Updated last year
- 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
- ML/DL Math and Method notes☆63Updated last year
- A library to create and manage configuration files, especially for machine learning projects.☆79Updated 3 years ago
- Fast Discounted Cumulative Sums in PyTorch☆96Updated 4 years ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆88Updated last year
- Amos optimizer with JEstimator lib.☆82Updated last year
- Helper scripts and notes that were used while porting various nlp models☆47Updated 3 years ago
- An implementation of Transformer with Expire-Span, a circuit for learning which memories to retain☆34Updated 4 years ago
- Implementation of TableFormer, Robust Transformer Modeling for Table-Text Encoding, in Pytorch☆39Updated 3 years ago
- Implementation of the Kalman Filtering Attention proposed in "Kalman Filtering Attention for User Behavior Modeling in CTR Prediction"☆58Updated last year
- My explorations into editing the knowledge and memories of an attention network☆35Updated 2 years ago
- Convert scikit-learn models to PyTorch modules☆165Updated last year
- a lightweight transformer library for PyTorch☆72Updated 3 years ago
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆32Updated 3 months ago
- ☆31Updated 3 months ago
- Various transformers for FSDP research☆38Updated 2 years 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 N-Grammer, augmenting Transformers with latent n-grams, in Pytorch☆76Updated 2 years ago
- ☆61Updated 3 years ago
- Implementation of Flash Attention in Jax☆218Updated last year
- Differentiable Algorithms and Algorithmic Supervision.☆116Updated 2 years ago
- JAX/Flax implementation of the Hyena Hierarchy☆34Updated 2 years ago
- Implementation of GateLoop Transformer in Pytorch and Jax☆90Updated last year
- Train very large language models in Jax.☆209Updated last year
- Simply Numpy implementation of the FAVOR+ attention mechanism, https://teddykoker.com/2020/11/performers/☆38Updated 4 years ago
- A place to store reusable transformer components of my own creation or found on the interwebs☆60Updated this week
- Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)☆189Updated 3 years ago
- A deep learning library based on Pytorch focussed on low resource language research and robustness☆70Updated 3 years ago