MatX-inc/seqax external links

---

GitHub - View repository

GitHub.dev - Open in online code editor

Star history - Growth chart over time

deps.dev - Dependencies and security vulnerabilities

Gitingest - Export source code for LLM usage

Repomix - Export source code for LLM usage

uithub - Export source code for LLM usage

Close

MatX-inc/seqax

Readme badge preview -

If you own this repo, copy the snippet below and add it to your README.md

---

Copy

[![RelatedRepos](https://img.shields.io/badge/related-repos-yellow)](https://relatedrepos.com/gh/MatX-inc/seqax)

Close

MatX-inc / seqaxView on GitHubMore

• External links
• Readme badge

seqax = sequence modeling + JAX

☆198Jul 23, 2025Updated last year

Alternatives and similar repositories for seqax

Users that are interested in seqax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.

• google-deepmind / nanodo

  View on GitHub
  ☆304Jul 15, 2024Updated 2 years ago
• yixiaoer / einshard

  View on GitHub
  Einsum-like high-level array sharding API for JAX
  ☆35Jul 16, 2024Updated 2 years ago
• thecharlieblake / lovely-llama

  View on GitHub
  An implementation of the Llama architecture, to instruct and delight
  ☆21May 31, 2025Updated last year
• marin-community / levanter

  View on GitHub
  Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax
  ☆708Jan 26, 2026Updated 6 months ago
• yixiaoer / mistral-v0.2-jax

  View on GitHub
  JAX implementation of the Mistral 7b v0.2 model
  ☆35Jul 3, 2024Updated 2 years ago
• AI Agents on DigitalOcean Gradient AI Platform • Ad
  Build production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
• young-geng / scalax

  View on GitHub
  A simple library for scaling up JAX programs
  ☆148Nov 4, 2025Updated 9 months ago
• young-geng / mintext

  View on GitHub
  Minimal but scalable implementation of large language models in JAX
  ☆34Nov 28, 2025Updated 8 months ago
• cloneofsimo / min-fsdp

  View on GitHub
  ☆93Jul 5, 2024Updated 2 years ago
• AI-Hypercomputer / maxtext

  View on GitHub
  A simple, performant, and scalable Jax LLM!
  ☆2,383Updated this week
• DramaCow / jaxued

  View on GitHub
  ☆99Jan 21, 2026Updated 6 months ago
• tensorpro / Autonomous-Braking

  View on GitHub
  Work towards a self-driving car's braking system with video demo.
  ☆10Dec 10, 2018Updated 7 years ago
• JonasGeiping / linear_cross_entropy_loss

  View on GitHub
  A fusion of a linear layer and a cross entropy loss, written for pytorch in triton.
  ☆75Aug 2, 2024Updated 2 years ago
• tensorpro / tpu_rwkv

  View on GitHub
  JAX implementations of RWKV
  ☆18Sep 26, 2023Updated 2 years ago
• kvfrans / jax-vqvae-vqgan

  View on GitHub
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆42Jun 6, 2024Updated 2 years ago
• Wordpress hosting with auto-scaling - Free Trial Offer • Ad
  Fully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
• FLAIROx / JaxGL

  View on GitHub
  Simple JAX Graphics Library.
  ☆38Nov 3, 2024Updated last year
• Sea-Snell / JAXSeq

  View on GitHub
  Train very large language models in Jax.
  ☆208Oct 21, 2023Updated 2 years ago
• davisyoshida / lorax

  View on GitHub
  LoRA for arbitrary JAX models and functions
  ☆143Feb 26, 2024Updated 2 years ago
• lucaslingle / mu_transformer

  View on GitHub
  Official implementation of 'A Large-Scale Exploration of mu-Transfer' (CoRR 2024)
  ☆31Jun 5, 2025Updated last year
• keraJLi / rejax

  View on GitHub
  Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!
  ☆274Jun 10, 2026Updated 2 months ago
• glassroom / heinsen_sequence

  View on GitHub
  Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)
  ☆98Dec 5, 2024Updated last year
• rwitten / HighPerfLLMs2024

  View on GitHub
  ☆591Jul 11, 2024Updated 2 years ago
• sekstini / basedxl

  View on GitHub
  ☆18Mar 18, 2024Updated 2 years ago
• ikostrikov / jaxrl2

  View on GitHub
  ☆58Jan 20, 2023Updated 3 years ago
• Wordpress hosting with auto-scaling - Free Trial Offer • Ad
  Fully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
• clankur / einygpt

  View on GitHub
  a transformer implemented primarily using einops and trained on the tinystories dataset
  ☆14Jun 21, 2024Updated 2 years ago
• Doraemonzzz / hgru-pytorch

  View on GitHub
  ☆29Jul 9, 2024Updated 2 years ago
• proger / nanokitchen

  View on GitHub
  Parallel Associative Scan for Language Models
  ☆18Jan 8, 2024Updated 2 years ago
• vvvm23 / mamba-jax

  View on GitHub
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆94Jan 25, 2024Updated 2 years ago
• nebius / kvax

  View on GitHub
  A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.
  ☆174Nov 11, 2025Updated 8 months ago
• instadeepai / flashbax

  View on GitHub
  ⚡ Flashbax: Accelerated Replay Buffers in JAX
  ☆280Sep 22, 2025Updated 10 months ago
• yixiaoer / mistral-jax

  View on GitHub
  JAX implementation of the Mistral 7b v0.1 model
  ☆13Mar 27, 2024Updated 2 years ago
• jax-ml / jax-llm-examples

  View on GitHub
  Minimal yet performant LLM examples in pure JAX
  ☆271Jul 4, 2026Updated last month
• yikangshen / megablocks

  View on GitHub
  ☆20May 30, 2024Updated 2 years ago
• Deploy to Railway using AI coding agents - Free Credits Offer • Ad
  Use Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
• google-deepmind / jmp

  View on GitHub
  JMP is a Mixed Precision library for JAX.
  ☆213Jul 8, 2026Updated last month
• Artur-Galstyan / statedict2pytree

  View on GitHub
  ☆47Feb 26, 2026Updated 5 months ago
• HazyResearch / train-tk

  View on GitHub
  train with kittens!
  ☆67Oct 25, 2024Updated last year
• google-deepmind / distrax

  View on GitHub
  ☆649Jul 30, 2026Updated last week
• jax-ml / jax-triton

  View on GitHub
  jax-triton contains integrations between JAX and OpenAI Triton
  ☆467Updated this week
• google / paxml

  View on GitHub
  Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…
  ☆558Updated this week
• google / aqt

  View on GitHub
  ☆360Updated this week