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:
- ☆34Updated 9 months ago
- ☆38Updated 11 months ago
- ☆18Updated last year
- ☆53Updated last year
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆45Updated last year
- Minimal Implimentation of VCRec (2024) for collapse provention.☆17Updated 5 months ago
- JAX Implementation of Black Forest Labs' Flux.1 family of models☆34Updated 8 months ago
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆17Updated 3 months ago
- ☆13Updated last week
- Generative cellular automaton-like learning environments for RL.☆19Updated 4 months ago
- Automatically take good care of your preemptible TPUs☆36Updated 2 years ago
- Latent Diffusion Language Models☆68Updated last year
- Simplex Random Feature attention, in PyTorch☆74Updated last year
- A JAX implementation of the continuous time formulation of Consistency Models☆85Updated 2 years ago
- ☆23Updated 6 months ago
- ☆19Updated last month
- ☆71Updated last year
- ☆49Updated last year
- Learning Universal Predictors☆76Updated 10 months ago
- ☆21Updated 7 months ago
- Train vision models using JAX and 🤗 transformers☆98Updated 2 months ago
- [WIP] Transformer to embed Danbooru labelsets☆13Updated last year
- research impl of Native Sparse Attention (2502.11089)☆54Updated 4 months ago
- RWKV model implementation☆38Updated last year
- ☆20Updated last year
- DiCE: The Infinitely Differentiable Monte-Carlo Estimator☆31Updated last year
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"☆101Updated 6 months ago
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆30Updated 3 weeks ago
- Experiment of using Tangent to autodiff triton☆79Updated last year
- ☆17Updated 10 months ago