Commits

Branch selector

 main

User selector

bvandermoon

Datepicker

All time

Commit History

Commits on Nov 4, 2025

• Use correct extra_deps_from_github.txt file path in setup.sh

  bvandermoon

  authored
  27eb9a9

  Copy full SHA for 27eb9a9

Commits on Nov 1, 2025

• Update Latest News MaxText installation link to new location

  bvandermoon

  authored
  3b485c4

  Copy full SHA for 3b485c4

Commits on Oct 31, 2025

• Alphabetize ReadTheDocs guides

  bvandermoon

  committed
  b68940a

  Copy full SHA for b68940a

Commits on Oct 30, 2025

• Move MaxText installation guide to separate ReadTheDocs page

  bvandermoon

  committed
  5e1b120

  Copy full SHA for 5e1b120

Commits on Oct 29, 2025

• Add user guide to update MaxText dependencies with seed-env

  bvandermoon

  committed
  73d2586

  Copy full SHA for 73d2586

Commits on Oct 17, 2025

• Include venv creation in README installation instructions

  Show description for 252cd4d
  bvandermoon

  authored
  252cd4d

  Copy full SHA for 252cd4d

Commits on Oct 10, 2025

• Update README.md

  Show description for 699595b
  bvandermoon

  committed
  699595b

  Copy full SHA for 699595b

Commits on Sep 29, 2025

• Continue pinning Tunix dependency for nightly builds

  

  bvandermoon

  authored and

  NicoGrande

  committed
  8d671b4

  Copy full SHA for 8d671b4

Commits on Sep 23, 2025

• Continue pinning Tunix dependency for nightly builds

  bvandermoon

  committed
  35d82e6

  Copy full SHA for 35d82e6

Commits on Sep 8, 2025

• Add src layout announcement

  bvandermoon

  committed
  eeee8cf

  Copy full SHA for eeee8cf

Commits on Sep 3, 2025

• Reverts 949215a209075da9f4fa9f82c7b9ee6522fa7822

  Show description for e1fde45
  

  bvandermoon

  authored and

  Google-ML-Automation

  committed
  e1fde45

  Copy full SHA for e1fde45

Commits on Aug 21, 2025

• Update RESTRUCTURE.md to include latest changes

  bvandermoon

  committed
  0d5db0c

  Copy full SHA for 0d5db0c

Commits on Aug 14, 2025

• Add to_linen_class for bridging decoder layers to Linen

  bvandermoon

  committed
  df42036

  Copy full SHA for df42036

Commits on Aug 13, 2025

• Reverts 5ccbf3b6fe5505170312c8cbf22106a05fcef2de

  Show description for a33a0fe
  

  bvandermoon

  authored and

  Google-ML-Automation

  committed
  a33a0fe

  Copy full SHA for a33a0fe

Commits on Aug 7, 2025

• Pass model_mode to decoder layer __init__ functions

  bvandermoon

  committed
  cceaca9

  Copy full SHA for cceaca9

Commits on Aug 6, 2025

• Accept inputs shape in Attention/MLA constructor

  bvandermoon

  committed
  dffa82b

  Copy full SHA for dffa82b

Commits on Aug 5, 2025

• Convert Attention and MLA modules to NNX

  bvandermoon

  committed
  8c5cb45

  Copy full SHA for 8c5cb45

Commits on Jul 29, 2025

• Fix variable name typo in KVCache

  bvandermoon

  committed
  21fc83c

  Copy full SHA for 21fc83c

Commits on Jul 23, 2025

• Migrate KVCache to NNX

  bvandermoon

  committed
  8e93f4e

  Copy full SHA for 8e93f4e

Commits on Jul 19, 2025

• Fix docker build issues - rollback PR 1971

  bvandermoon

  committed
  27182d7

  Copy full SHA for 27182d7

Commits on Jul 17, 2025

• Request feedback for proposed repo restructuring

  bvandermoon

  committed
  85dc8ae

  Copy full SHA for 85dc8ae

Commits on Jul 11, 2025

• Migrate PagedAttentionOp to use NNX

  bvandermoon

  committed
  cc7da1c

  Copy full SHA for cc7da1c

Commits on Jul 9, 2025

• Migrate Embed module to NNX

  bvandermoon

  committed
  31af45d

  Copy full SHA for 31af45d

Commits on Jul 2, 2025

• Migrate YarnRotaryEmbedding, LlamaVisionRotaryEmbedding, and PositionalEmbedding to NNX

  bvandermoon

  committed
  e78827a

  Copy full SHA for e78827a

Commits on Jun 30, 2025

• Add name to linen modules for RotaryEmbedding and LLaMARotaryEmbedding

  bvandermoon

  committed
  7e2afeb

  Copy full SHA for 7e2afeb

Commits on Jun 27, 2025

• Migrate RotaryEmbedding and LLaMARotaryEmbedding to NNX

  bvandermoon

  committed
  790c325

  Copy full SHA for 790c325

Commits on Jun 26, 2025

• Pass metadata to Linen for AttentionOp and L2Norm

  bvandermoon

  committed
  1aa1163

  Copy full SHA for 1aa1163

• Remove unused setup.py files

  bvandermoon

  committed
  f383d57

  Copy full SHA for f383d57

Commits on Jun 24, 2025

• Migrate AttentionOp to use NNX

  bvandermoon

  committed
  6d021dc

  Copy full SHA for 6d021dc

Commits on Jun 20, 2025

• Update L2Norm module to use NNX

  bvandermoon

  committed
  c2d40fe

  Copy full SHA for c2d40fe

Commits on Jun 18, 2025

• Update DenseGeneral in_feature and feature arg names

  bvandermoon

  committed
  fff7083

  Copy full SHA for fff7083

Commits on Jun 5, 2025

• Convert DenseGeneral to NNX

  Show description for 9982c45
  

  bvandermoon

  and

  Cristian Garcia

  committed
  9982c45

  Copy full SHA for 9982c45

Commits on May 13, 2025

• Add test comment to only run grpo_trainer_correctness_test on v4-8

  bvandermoon

  committed
  a7270fd

  Copy full SHA for a7270fd

Commits on May 9, 2025

• Remove hard-coded full dependency set for JAX AI image 0.5.2

  bvandermoon

  committed
  a664648

  Copy full SHA for a664648

Commits on May 8, 2025

• Fix MaxText pyconfig import in page_manager_test

  bvandermoon

  committed
  59e9778

  Copy full SHA for 59e9778

Pagination

Previous

Next