EleutherAI / training-jacobianLinks
☆23Updated 6 months ago
Alternatives and similar repositories for training-jacobian
Users that are interested in training-jacobian are comparing it to the libraries listed below
Sorting:
- Minimum Description Length probing for neural network representations☆18Updated 4 months ago
- Implementation of Spectral State Space Models☆16Updated last year
- Code for the paper "Function-Space Learning Rates"☆20Updated 3 weeks ago
- ☆18Updated last year
- Engineering the state of RNN language models (Mamba, RWKV, etc.)☆32Updated last year
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆17Updated 3 months ago
- ☆19Updated last month
- Code for the examples presented in the talk "Training a Llama in your backyard: fine-tuning very large models on consumer hardware" given…☆14Updated last year
- DiCE: The Infinitely Differentiable Monte-Carlo Estimator☆31Updated last year
- ☆34Updated 9 months ago
- ☆21Updated 7 months ago
- Triton Implementation of HyperAttention Algorithm☆48Updated last year
- Combining SOAP and MUON☆16Updated 4 months ago
- Fork of Flame repo for training of some new stuff in development☆14Updated last week
- Jax like function transformation engine but micro, microjax☆32Updated 7 months ago
- This repo is based on https://github.com/jiaweizzhao/GaLore☆28Updated 9 months ago
- Understanding how features learned by neural networks evolve throughout training☆35Updated 8 months ago
- Latent Diffusion Language Models☆68Updated last year
- Implementation of a holodeck, written in Pytorch☆18Updated last year
- My explorations into editing the knowledge and memories of an attention network☆35Updated 2 years ago
- Codes accompanying the paper "LaProp: a Better Way to Combine Momentum with Adaptive Gradient"☆29Updated 4 years ago
- [NeurIPS 2023] Sparse Modular Activation for Efficient Sequence Modeling☆37Updated last year
- Implementation of Token Shift GPT - An autoregressive model that solely relies on shifting the sequence space for mixing☆50Updated 3 years ago
- ☆53Updated last year
- Lottery Ticket Adaptation☆39Updated 7 months ago
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆30Updated 2 weeks ago
- GoldFinch and other hybrid transformer components☆45Updated 11 months ago
- RWKV model implementation☆38Updated last year
- Experimental scripts for researching data adaptive learning rate scheduling.☆23Updated last year
- Demonstration that finetuning RoPE model on larger sequences than the pre-trained model adapts the model context limit☆63Updated 2 years ago