A zero-to-one guide on scaling modern transformers with n-dimensional parallelism.
☆119Dec 29, 2025Updated 3 months ago
Alternatives and similar repositories for JAXformer
Users that are interested in JAXformer are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Tidy autoregressive inference in JAX☆15Sep 1, 2025Updated 6 months ago
- Minimal (truly) muP implementation, consistent with TP4 and TP5 papers notation☆14Jan 2, 2026Updated 2 months ago
- Custom triton kernels for training Karpathy's nanoGPT.☆19Oct 21, 2024Updated last year
- ☆50Mar 14, 2025Updated last year
- ☆77Feb 18, 2026Updated last month
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Fast, simple, cryptographically strong random numbers in C++. Experimental.☆19Dec 12, 2013Updated 12 years ago
- Minimal but scalable implementation of large language models in JAX☆35Nov 28, 2025Updated 4 months ago
- Martingale posterior neural networks for fast sequential decision making @ Neurips 2025☆23Nov 13, 2025Updated 4 months ago
- Website for CSE 234, Winter 2025☆13Mar 24, 2025Updated last year
- A repository to unravel the language of GPUs, making their kernel conversations easy to understand☆201Jun 1, 2025Updated 9 months ago
- Frechet inception distance (FID) evaluation in JAX☆14May 28, 2024Updated last year
- A JAX implementation of stochastic addition.☆14Aug 15, 2022Updated 3 years ago
- Tensor Parallelism with JAX + Shard Map☆11Sep 29, 2023Updated 2 years ago
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆19Jul 24, 2025Updated 8 months ago
- Simple, predictable pricing with DigitalOcean hosting • AdAlways know what you'll pay with monthly caps and flat pricing. Enterprise-grade infrastructure trusted by 600k+ customers.
- Codebase for Linguistic Collapse: Neural Collapse in (Large) Language Models [NeurIPS 2024] [arXiv:2405.17767]☆18Apr 14, 2025Updated 11 months ago
- An efficient implementation of the NSA (Native Sparse Attention) kernel☆132Jun 24, 2025Updated 9 months ago
- Contains JAX implementation of algorithms for inverse reinforcement learning☆75Aug 18, 2024Updated last year
- Optimize the construction of earthquake-resistant buildings☆10Jul 7, 2024Updated last year
- ☆29Dec 15, 2025Updated 3 months ago
- Get a hack club dino☆12Mar 13, 2023Updated 3 years ago
- Browse Lance tables from your local machine in a simple web UI. No database to set up. Mount a folder and go.☆26Mar 16, 2026Updated last week
- This is the code corresponding to our publication introducing ConvDecoder with physics-based regularization (CD+r) for MRI☆10Feb 6, 2026Updated last month
- Efficiently discovering algorithms via LLMs with evolutionary search and reinforcement learning.☆17Apr 22, 2025Updated 11 months ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Code and report for APMA136 Final Project☆19May 6, 2015Updated 10 years ago
- [ICML 2025] Repository for M3-JEPA: Multimodal Alignment via Multi-gate MoE based on the Joint-Predictive Embedding Architecture☆23Mar 13, 2026Updated 2 weeks ago
- Hack Club Bank CLI☆10Jul 25, 2022Updated 3 years ago
- Community Eventing and Scripting examples☆19Aug 11, 2025Updated 7 months ago
- ⌨️ the simplest and smallest code editor for web, with no dependencies - forked from spell 🪄☆14Nov 5, 2025Updated 4 months ago
- Next.js app to display the word count of given text.☆12Mar 5, 2023Updated 3 years ago
- 🦘Websites upside down for those down under!☆13Nov 20, 2021Updated 4 years ago
- ☆10Apr 21, 2025Updated 11 months ago
- Utility to use eleven lab's streaming to in the command line☆11Aug 8, 2023Updated 2 years ago
- Wordpress hosting with auto-scaling on Cloudways • AdFully Managed hosting built for WordPress-powered businesses that need reliable, auto-scalable hosting. Cloudways SafeUpdates now available.
- TPU pod commander is a package for managing and launching jobs on Google Cloud TPU pods.☆21Sep 24, 2025Updated 6 months ago
- A reliable leaderboard algorithm for machine learning competitions☆17May 19, 2015Updated 10 years ago
- Numerical relativity surrogate waveform in Jax☆19Aug 14, 2025Updated 7 months ago
- ☆13Jul 12, 2024Updated last year
- implementation of https://arxiv.org/pdf/2312.09299☆21Jul 3, 2024Updated last year
- Twitter scraping library☆11Nov 11, 2021Updated 4 years ago
- Bayesian Deep Ensembles via MILE: easy to use, scikit-learn compatible and fast (JAX powered)☆42Mar 13, 2026Updated 2 weeks ago