teddykoker / tinyloaderLinks
☆68Updated 10 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 2 years ago
- A case study of efficient training of large language models using commodity hardware.☆68Updated 3 years ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆92Updated 2 years ago
- HomebrewNLP in JAX flavour for maintable TPU-Training☆51Updated 2 years ago
- ☆31Updated 2 weeks ago
- JAX/Flax implementation of the Hyena Hierarchy☆34Updated 2 years ago
- Convert scikit-learn models to PyTorch modules☆168Updated last year
- Amos optimizer with JEstimator lib.☆82Updated last year
- Functional deep learning☆108Updated 3 years ago
- A library to create and manage configuration files, especially for machine learning projects.☆79Updated 3 years ago
- Automatically take good care of your preemptible TPUs☆37Updated 2 years ago
- A place to store reusable transformer components of my own creation or found on the interwebs☆72Updated 3 weeks ago
- Machine Learning eXperiment Utilities☆48Updated 6 months ago
- Implementation of GateLoop Transformer in Pytorch and Jax☆92Updated last year
- Implementation of TableFormer, Robust Transformer Modeling for Table-Text Encoding, in Pytorch☆39Updated 3 years ago
- A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-…☆66Updated 2 years ago
- Fast Discounted Cumulative Sums in PyTorch☆97Updated 4 years ago
- My explorations into editing the knowledge and memories of an attention network☆35Updated 3 years ago
- This repository contains example code to build models on TPUs☆30Updated 2 years ago
- LayerNorm(SmallInit(Embedding)) in a Transformer to improve convergence☆61Updated 3 years ago
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆47Updated last year
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆32Updated 8 months ago
- Hyperparameter tuning via uncertainty modeling☆49Updated last year
- A framework for implementing equivariant DL☆10Updated 4 years ago
- An implementation of Transformer with Expire-Span, a circuit for learning which memories to retain☆34Updated 5 years ago
- Explores the ideas presented in Deep Ensembles: A Loss Landscape Perspective (https://arxiv.org/abs/1912.02757) by Stanislav Fort, Huiyi …☆66Updated 5 years ago
- a lightweight transformer library for PyTorch☆72Updated 4 years ago
- Various transformers for FSDP research☆38Updated 3 years ago
- Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)☆190Updated 3 years ago
- Unofficial JAX implementations of deep learning research papers☆161Updated 3 years ago