☆16Oct 20, 2025Updated 4 months ago
Alternatives and similar repositories for drjax
Users that are interested in drjax are comparing it to the libraries listed below
Sorting:
- ☆12Jun 2, 2023Updated 2 years ago
- Source-to-Source Debuggable Derivatives in Pure Python☆15Jan 23, 2024Updated 2 years ago
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆18Jul 24, 2025Updated 7 months ago
- Fork of Flame repo for training of some new stuff in development☆19Feb 20, 2026Updated last week
- Simple python library for generating your own perfetto traces for your application. Can be used for both app instrumentation and custom …☆25Jun 22, 2025Updated 8 months ago
- An implementation of the Llama architecture, to instruct and delight☆21May 31, 2025Updated 9 months ago
- ☆23Jan 5, 2025Updated last year
- ☆61Updated this week
- ☆23Jun 18, 2024Updated last year
- Efficient encoder-decoder architecture for small language models (≤1B parameters) with cross-architecture knowledge distillation and visi…☆33Feb 7, 2025Updated last year
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆32Jun 5, 2025Updated 8 months ago
- ☆34Sep 10, 2024Updated last year
- Implementation of PSGD optimizer in JAX☆35Dec 31, 2024Updated last year
- A comprehensive JAX/NNX library for diffusion and flow matching generative algorithms, featuring DiT (Diffusion Transformer) and its vari…☆137Oct 16, 2025Updated 4 months ago
- Repository for go shared libraries (for now).☆11Dec 1, 2025Updated 3 months ago
- Using FlexAttention to compute attention with different masking patterns☆47Sep 22, 2024Updated last year
- Open-source Human Feedback Library☆11Oct 25, 2023Updated 2 years ago
- Multimodal SuperCon: Classifier for Drivers of Deforestation in Indonesia☆10Nov 18, 2023Updated 2 years ago
- ☆12Jan 5, 2026Updated last month
- ☆10Oct 24, 2023Updated 2 years ago
- League of Legends - Champion Database, Items, Spells, Abilities, Runes, Lore, Portraits and Stats.☆10Nov 9, 2012Updated 13 years ago
- Framework for geospatial research CNN development.☆11Apr 10, 2021Updated 4 years ago
- ☆14Jan 23, 2025Updated last year
- Two-stream remote sensing model for water quality mapping: 2SeaColor☆10Feb 2, 2021Updated 5 years ago
- Code for reproducing the results from "CrAM: A Compression-Aware Minimizer" accepted at ICLR 2023☆10Mar 1, 2023Updated 3 years ago
- ☆292Jul 15, 2024Updated last year
- Code generation for a statically typed bridge between the client and server in TypeScript☆10Dec 4, 2017Updated 8 years ago
- ☆13Jun 18, 2024Updated last year
- 4-bit Shampoo for Memory-Efficient Network Training (NeurIPS 2024)☆13Feb 13, 2025Updated last year
- ☆13Jun 22, 2025Updated 8 months ago
- 🚀 Collection of libraries used with fms-hf-tuning to accelerate fine-tuning and training of large models.☆13Jan 30, 2026Updated last month
- Multi-heap-sort for many small arrays, quicksort with 3 pivots for one big array, CUDA acceleration, CUDA memory compression.☆13Sep 29, 2024Updated last year
- An implementation of the Augmented Random Search algorithm☆14Jan 29, 2022Updated 4 years ago
- ☆11Jul 21, 2024Updated last year
- My personal repository☆11Jan 6, 2024Updated 2 years ago
- Trying Tigerbeetle transactional database.☆11Jul 14, 2024Updated last year
- The Word Embedding Database API☆11Aug 20, 2019Updated 6 years ago
- Clustered Compositional Embeddings☆11Oct 25, 2023Updated 2 years ago
- ☆10Jun 1, 2022Updated 3 years ago