yk / llmvmLinks
☆30Updated last year
Alternatives and similar repositories for llmvm
Users that are interested in llmvm are comparing it to the libraries listed below
Sorting:
- ☆53Updated last year
- Learning Universal Predictors☆78Updated last year
- RWKV model implementation☆38Updated 2 years ago
- ☆100Updated 2 weeks ago
- gzip Predicts Data-dependent Scaling Laws☆35Updated last year
- ☆27Updated last year
- ☆38Updated last year
- ☆34Updated 11 months ago
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"☆101Updated 7 months ago
- DeMo: Decoupled Momentum Optimization☆190Updated 8 months ago
- Experiment of using Tangent to autodiff triton☆80Updated last year
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆85Updated last year
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆45Updated last year
- ☆18Updated last year
- Jax like function transformation engine but micro, microjax☆33Updated 9 months ago
- Simplex Random Feature attention, in PyTorch☆74Updated last year
- Generative cellular automaton-like learning environments for RL.☆19Updated 6 months ago
- Latent Diffusion Language Models☆69Updated last year
- train with kittens!☆62Updated 9 months ago
- Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…☆71Updated last month
- Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)☆94Updated 8 months ago
- ☆23Updated 8 months ago
- ☆12Updated last week
- Automatically take good care of your preemptible TPUs☆36Updated 2 years ago
- ☆61Updated 3 years ago
- ☆62Updated 8 months ago
- ☆71Updated last year
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆46Updated last year
- A programming language for formal/informal computation.☆41Updated 2 weeks ago
- Understanding how features learned by neural networks evolve throughout training☆36Updated 9 months ago