AakashKumarNain / nanoGPTJAXView external linksLinks
Implementing scalable LLMs in pure JAX (no third-party libraries)
☆43Updated this week
Alternatives and similar repositories for nanoGPTJAX
Users that are interested in nanoGPTJAX are comparing it to the libraries listed below
Sorting:
- This is a port of Mistral-7B model in JAX☆33Jul 1, 2024Updated last year
- ☆35Feb 5, 2024Updated 2 years ago
- Diffusion Probabilistic Model in Jax☆11Apr 20, 2024Updated last year
- Applications leveraged by Statistics Canada, ready to launch on Kubernetes using Helm.☆10Dec 8, 2025Updated 2 months ago
- Topology library for Coq☆12Dec 24, 2015Updated 10 years ago
- ☆14Sep 29, 2025Updated 4 months ago
- General framework for Bayesian inversion of continuous hierarchical models☆10Sep 20, 2021Updated 4 years ago
- A clean no-jargon mathematical definition of transforrmer language model with a Python implementation that focuses on clarity rather than…☆11Jul 23, 2022Updated 3 years ago
- ☆39Jan 27, 2026Updated 2 weeks ago
- A minimal toy async-await executor in web assembly☆11Oct 8, 2019Updated 6 years ago
- Minimalist version of probml/rebayes☆10Sep 15, 2025Updated 4 months ago
- An implementation of several unsupervised object discovery models (Slot Attention, SLATE, GNM) in PyTorch with pre-trained models.☆14May 26, 2025Updated 8 months ago
- ☆14Mar 9, 2023Updated 2 years ago
- ☆19Nov 25, 2024Updated last year
- Collection of ASR models for English TFLite models for faster inference.☆13Feb 21, 2022Updated 3 years ago
- Vehicle control☆11Jun 8, 2019Updated 6 years ago
- msgpack-javascript ported to deno☆13Jun 13, 2021Updated 4 years ago
- Sequential Monte Carlo sampler for PyMC2 models.☆13Apr 4, 2018Updated 7 years ago
- Code for the paper "Closing the Curious Case of Neural Text Degeneration"☆11Apr 9, 2025Updated 10 months ago
- An implementation of DreamerV2 written in JAX, with support for running multiple random seeds of an experiment on a single GPU.☆18Jan 16, 2023Updated 3 years ago
- [ICCV 2021] Click to Move: Controlling Video Generation with Sparse Motion☆11Apr 14, 2023Updated 2 years ago
- Disjoint borrow of slices in Rust☆11Sep 1, 2019Updated 6 years ago
- The BitTorrent lt_donthave extension (BEP 54)☆13Jan 30, 2026Updated 2 weeks ago
- Bayesian model reduction for probabilistic machine learning☆11Jul 3, 2025Updated 7 months ago
- A library supporting efficient stereogram image computation.☆12Jun 2, 2022Updated 3 years ago
- Handle Xz as easily as Zlib☆10Updated this week
- AlphaRING is a package designed for interpretable, protein structure-based prediction of missense variant deleteriousness.☆22Aug 6, 2025Updated 6 months ago
- GraphQL WS with Multi GraphQL servers☆14Oct 22, 2020Updated 5 years ago
- ☆14Jan 15, 2026Updated 3 weeks ago
- dcEmb, the Embecosm Dynamic Causal Modelling library☆13Aug 26, 2024Updated last year
- ☆22Dec 19, 2025Updated last month
- ☆12Mar 18, 2022Updated 3 years ago
- Perturbational analysis by causality-aware generative model for single-cell RNA-sequencing data☆19Dec 16, 2025Updated last month
- Open-source MATLAB to Python Compiler (from 2008)☆15Jun 18, 2025Updated 7 months ago
- Code for Fooling Contrastive Language-Image Pre-trainined Models with CLIPMasterPrints☆15Jan 25, 2026Updated 2 weeks ago
- ☆20Oct 22, 2025Updated 3 months ago
- Variant impact phenotyping using Perturb-seq☆10Apr 22, 2024Updated last year
- Open source Java framework to create, process and manage mixtures of exponential family☆14Aug 4, 2015Updated 10 years ago
- ☆20Jan 10, 2026Updated last month