Commits

Breadcrumbs

History for

onmain

User selector

All users

All time

Commit History

Commits on Jun 1, 2026

• Fixed gemma example sampler test test_models_from_kaggle

  vfdev-5

  authored
  ec75218

  Copy full SHA for ec75218

Commits on Apr 23, 2026

• Merge pull request #5427 from google:update-gemma-docs

  Show description for 7075c6e
  

  Flax Authors

  committed
  7075c6e

  Copy full SHA for 7075c6e

• Rename vars_as to with_vars

  samanklesaria

  committed
  d07091f

  Copy full SHA for d07091f

• Updated gemma docs

  vfdev-5

  committed
  b788d79

  Copy full SHA for b788d79

Commits on Apr 20, 2026

• Addressed comments

  vfdev-5

  authored
  58658ac

  Copy full SHA for 58658ac

Commits on Apr 15, 2026

• Fixed intermedites capture

  vfdev-5

  committed
  cdfadd6

  Copy full SHA for cdfadd6

Commits on Apr 13, 2026

• Merge pull request #4913 from google:gemma-example-using-grain

  Show description for 9bed435
  

  Flax Authors

  committed
  9bed435

  Copy full SHA for 9bed435

Commits on Apr 9, 2026

• Avoid deprecated pattern of constructing a dtype from an array.

  Show description for 9534e26
  

  Jake VanderPlas

  authored and

  Flax Authors

  committed
  9534e26

  Copy full SHA for 9534e26

Commits on Mar 26, 2026

• More Code updates

  Show description for 5260a1e
  vfdev-5

  committed
  5260a1e

  Copy full SHA for 5260a1e

Commits on Mar 25, 2026

• Added mixed precision: computation and weights dtypes

  vfdev-5

  committed
  cdf3b0c

  Copy full SHA for cdf3b0c

• Use grain in gemma example

  Show description for 199e653
  vfdev-5

  committed
  199e653

  Copy full SHA for 199e653

Commits on Mar 12, 2026

• Remove jax/tools/colab_tpu.py.

  Show description for 8ed67b8
  

  hawkinsp

  authored and

  Flax Authors

  committed
  8ed67b8

  Copy full SHA for 8ed67b8

Commits on Mar 2, 2026

• No public description

  Show description for 5486a63
  

  danielsuo

  authored and

  Flax Authors

  committed
  5486a63

  Copy full SHA for 5486a63

Commits on Feb 12, 2026

• Rename sharding_names to out_sharding in NNX Variable metadata

  Show description for a5c7683
  

  Cristian Garcia

  authored and

  Flax Authors

  committed
  a5c7683

  Copy full SHA for a5c7683

Commits on Feb 9, 2026

• chore: Migrate gsutil usage to gcloud storage

  gurusai-voleti

  committed
  5d7e13a

  Copy full SHA for 5d7e13a

Commits on Jan 13, 2026

• Update mnist example to use NNX.

  

  danielsuo

  authored and

  samanklesaria

  committed
  0ef9abb

  Copy full SHA for 0ef9abb

Commits on Jan 9, 2026

• Copybara import of the project:

  Show description for a7929b3
  

  Flax Team

  committed
  a7929b3

  Copy full SHA for a7929b3

Commits on Jan 8, 2026

• Merge pull request #5089 from samanklesaria:sharding_metadata

  Show description for 2863e25
  

  Flax Authors

  committed
  2863e25

  Copy full SHA for 2863e25

Commits on Jan 6, 2026

• improve hijax guide

  cgarciae

  committed
  5e31fe9

  Copy full SHA for 5e31fe9

• [pmap] Avoid degraded performance under the new `jax.pmap`.

  Show description for 5d90851
  

  danielsuo

  authored and

  Flax Authors

  committed
  5d90851

  Copy full SHA for 5d90851

Commits on Jan 5, 2026

• Rename sharding_names to sharding_metadata

  samanklesaria

  committed
  3e96343

  Copy full SHA for 3e96343

Commits on Dec 18, 2025

• Release 0.12.2

  IvyZX

  committed
  6d36623

  Copy full SHA for 6d36623

Commits on Dec 5, 2025

• [flax:examples:nlp_seq] Create a main.py file to run tests with config files to match other examples. #jax-fixit

  Show description for 872d50b
  

  danielsuo

  authored and

  Flax Authors

  committed
  872d50b

  Copy full SHA for 872d50b

• [flax:examples:lm1b_nnx] Update example to work internally. #jax-fixit.

  Show description for 881b8ed
  

  danielsuo

  authored and

  Flax Authors

  committed
  881b8ed

  Copy full SHA for 881b8ed

Commits on Dec 3, 2025

• [flax:examples:ppo] Fix some linter / import issues. #jax-fixit

  Show description for 59cfd99
  

  danielsuo

  authored and

  Flax Authors

  committed
  59cfd99

  Copy full SHA for 59cfd99

Commits on Dec 2, 2025

• [flax:examples:sst2] Fix notebook error.

  Show description for 5a3d148
  

  danielsuo

  authored and

  Flax Authors

  committed
  5a3d148

  Copy full SHA for 5a3d148

Commits on Dec 1, 2025

• [flax:examples:sst2] Fix pytype errors.

  Show description for 6ac5a78
  

  danielsuo

  authored and

  Flax Authors

  committed
  6ac5a78

  Copy full SHA for 6ac5a78

• [flax:examples:gemma] Fixing linter errors.

  Show description for 324f4b4
  

  danielsuo

  authored and

  Flax Authors

  committed
  324f4b4

  Copy full SHA for 324f4b4

• [flax:examples:vae] Small linter fixes.

  Show description for 88d7497
  

  danielsuo

  authored and

  Flax Authors

  committed
  88d7497

  Copy full SHA for 88d7497

• [flax:examples:seq2seq] Create main and default config based on seq2seq.ipynb.

  Show description for 9ed59e2
  

  danielsuo

  authored and

  Flax Authors

  committed
  9ed59e2

  Copy full SHA for 9ed59e2

• [flax:examples:wmt] Small linter fixes.

  Show description for c29b6d6
  

  danielsuo

  authored and

  Flax Authors

  committed
  c29b6d6

  Copy full SHA for c29b6d6

Commits on Nov 17, 2025

• deprecate .value

  cgarciae

  committed
  a2eac32

  Copy full SHA for a2eac32

Commits on Nov 14, 2025

• add HijaxVariable [rollforward]

  Show description for b5db513
  

  Cristian Garcia

  authored and

  Flax Authors

  committed
  b5db513

  Copy full SHA for b5db513

Commits on Nov 8, 2025

• Copybara import of the project:

  Show description for 8e9f8ae
  

  Flax Team

  committed
  8e9f8ae

  Copy full SHA for 8e9f8ae

Commits on Nov 7, 2025

• Copybara import of the project:

  Show description for 18063be
  

  cgarciae

  authored and

  Flax Authors

  committed
  18063be

  Copy full SHA for 18063be