yk / llmvm
☆30Updated last year
Alternatives and similar repositories for llmvm:
Users that are interested in llmvm are comparing it to the libraries listed below
- ☆33Updated 6 months ago
- ☆21Updated 3 months ago
- ☆27Updated 8 months ago
- ☆49Updated last year
- ☆53Updated last year
- ☆21Updated 4 months ago
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆17Updated 2 weeks ago
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆103Updated 4 months ago
- Simplex Random Feature attention, in PyTorch☆74Updated last year
- Demonstration that finetuning RoPE model on larger sequences than the pre-trained model adapts the model context limit☆63Updated last year
- ☆22Updated last year
- ☆92Updated 2 months ago
- Make triton easier☆47Updated 9 months ago
- JAX Implementation of Black Forest Labs' Flux.1 family of models☆30Updated 5 months ago
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"☆98Updated 3 months ago
- RWKV model implementation☆37Updated last year
- ☆38Updated 8 months ago
- research impl of Native Sparse Attention (2502.11089)☆53Updated last month
- Latent Diffusion Language Models☆68Updated last year
- [WIP] Transformer to embed Danbooru labelsets☆13Updated last year
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆30Updated 3 months ago
- Experiment of using Tangent to autodiff triton☆78Updated last year
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆44Updated 10 months ago
- ☆60Updated 3 years ago
- Automatically take good care of your preemptible TPUs☆36Updated last year
- Implementation of Spectral State Space Models☆16Updated last year
- Jax like function transformation engine but micro, microjax☆30Updated 5 months ago
- train with kittens!☆55Updated 5 months ago
- Learning Universal Predictors☆73Updated 8 months ago
- Code to reproduce "Transformers Can Do Arithmetic with the Right Embeddings", McLeish et al (NeurIPS 2024)☆187Updated 10 months ago