All pull requests

Pull requests

• Add Alphafold to GHCR registry

  #2375·Steboss opened Oct 9, 2026Contributor·

  ·
  1

• Unpin huggingface-hub in JAX inference offloading workflow

  #2372·aybchan opened Oct 8, 2026Member·

  ·
  3

• Revert "fix versions for tf and tf-text to prevent bug (#2287)"

  #2331·aybchan opened Sep 17, 2026Member·

  ·
  3

• docs: add DeepSeek-v3 MoE reference configuration for MaxText

  #2322·shljessie opened Sep 10, 2026Contributor·2 of 3 tasks·

  ·
  1

• alphafold3: bump tokamax pin

  #2321·olupton opened Sep 10, 2026Collaborator·

  ·
  3

• Update 26.09-devel from main

  #2315·aybchan opened Sep 8, 2026Member·

  ·
  1

• --commit feature to triage tool

  #2298·Steboss opened Aug 28, 2026Contributor·

  ·
  7

• Add jax-cudnn-frontend agent skill

  #2273·jessicadesilva opened Aug 17, 2026Member·

  ·
  8

• Remove staging image docs

  #2252·aybchan opened Aug 4, 2026Member·

  ·
  2

• Improve build-te for build-jax step in CI

  #2245·Steboss opened Jul 30, 2026Contributor·

  ·
  25

• Document scheduler memory fencing flags

  #2242·shawnwang18 opened Jul 29, 2026·

  ·
  1

• General cleaning and fixing of ci

  #2213·Steboss opened Jul 9, 2026Contributor·

  ·
  5

• Set to load-balance jobs on GKE

  #2151·aybchan opened Jun 5, 2026Member·

  ·
  3

• Refactor of vLLM bridge

  #1914·keshavb96 opened Jan 23, 2026Contributor·

  ·
  7

• Install uv (https://docs.astral.sh/uv/) in base container

  #1891·andportnoy opened Jan 12, 2026Contributor·

  ·

• [jax-inference-offloading] consolidate definitions for default tensor dtype

  #1816·yhtang opened Dec 5, 2025Contributor·

  ·
  8

• [jax-inference-offloading] Bump to CUDA 13.0

  #1807·yhtang opened Dec 2, 2025Contributor·

  ·

• Add MACE training example

  #1192·mariogeiger opened Dec 10, 2024·

  ·

• Customize NCCL for base container

  #1123·DwarKapex opened Oct 28, 2024Contributor·

  ·
  2

• ci[maxtext]: adds rosetta directory to maxtext containers

  #1097·terrykong opened Oct 13, 2024Contributor·

  ·
  2

• chore: remove deprecated flags from model xla flag configs

  #1088·terrykong opened Oct 8, 2024Contributor·

  ·
  1

• Fix the Order of GatedAct to be (act, linear) when Pax2TE.

  #892·mingxu1067 opened Jun 10, 2024Contributor·

  ·

• Add upstream pax 2-node Grok test

  #854·ashors1 opened May 24, 2024Contributor·

  ·
  2

• Support auto-generated FP8 meta for CKPT converters.

  #844·mingxu1067 opened May 21, 2024Contributor·

  ·
  4

• Add FMHA PAXML test

  #830·hmonishN opened May 16, 2024Contributor·

  ·
  33