Commits

Branch selector

 main

User selector

All users

All time

Commit History

Commits on Jun 19, 2026

• update to v0.12.8

  Show description for 3a49110
  

  Cristian Garcia

  authored and

  Flax Authors

  committed
  3a49110

  Copy full SHA for 3a49110

Commits on Jun 17, 2026

• optimize jit_partial

  Show description for 0a0aca7
  

  Cristian Garcia

  authored and

  Flax Authors

  committed
  0a0aca7

  Copy full SHA for 0a0aca7

• Merge pull request #5507 from google:add-dependabot

  Show description for 0682fe4
  

  Flax Authors

  committed
  0682fe4

  Copy full SHA for 0682fe4

• Merge pull request #5506 from google:fix-ci-nest-asyncio

  Show description for 6fec30f
  

  Flax Authors

  committed
  6fec30f

  Copy full SHA for 6fec30f

• Fix CI

  Show description for 5768bb5
  vfdev-5

  committed
  5768bb5

  Copy full SHA for 5768bb5

• Add Dependabot configuration for dependency updates

  Show description for 0a68232
  vfdev-5

  authored
  0a68232

  Copy full SHA for 0a68232

Commits on Jun 16, 2026

• Merge pull request #5455 from google:docs-examples--vit

  Show description for e48ce36
  

  Flax Authors

  committed
  e48ce36

  Copy full SHA for e48ce36

• Merge pull request #5492 from mohsinm-dev:fix-multimetric-hashable-graphdef

  Show description for 4c59d8f
  

  Flax Authors

  committed
  4c59d8f

  Copy full SHA for 4c59d8f

Commits on Jun 11, 2026

• fix MultiMetric GraphDef not hashable under jit (#5489)

  Show description for 6f74592
  mohsinm-dev

  committed
  6f74592

  Copy full SHA for 6f74592

Commits on Jun 5, 2026

• Merge pull request #5477 from google:fix-gemma-sampler-test-kaggle

  Show description for 5ab9d14
  

  Flax Authors

  committed
  5ab9d14

  Copy full SHA for 5ab9d14

• Merge branch 'main' into docs-examples--vit

  vfdev-5

  authored
  4ad282f

  Copy full SHA for 4ad282f

Commits on Jun 2, 2026

• Merge pull request #5463 from google:docs-examples--unetr

  Show description for 77a603b
  

  Flax Authors

  committed
  77a603b

  Copy full SHA for 77a603b

• Added image segmentation example

  Show description for 094b1a1
  vfdev-5

  committed
  094b1a1

  Copy full SHA for 094b1a1

• Added ViT training example

  vfdev-5

  committed
  a450307

  Copy full SHA for a450307

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 May 22, 2026

• optimize transform updates

  Show description for 0f128b5
  

  Cristian Garcia

  authored and

  Flax Authors

  committed
  0f128b5

  Copy full SHA for 0f128b5

Commits on May 20, 2026

• add storage backend type in saving telemetry.

  Show description for 218c7ff
  

  lyglst

  authored and

  Flax Authors

  committed
  218c7ff

  Copy full SHA for 218c7ff

Commits on May 19, 2026

• Merge pull request #5454 from samanklesaria:dataloader_docs

  Show description for 8485ec3
  

  Flax Authors

  committed
  8485ec3

  Copy full SHA for 8485ec3

Commits on May 15, 2026

• Merge pull request #5363 from samanklesaria:rng_guide_update

  Show description for d044de8
  

  Flax Authors

  committed
  d044de8

  Copy full SHA for d044de8

• Add data loaders tutorial

  samanklesaria

  committed
  64145c4

  Copy full SHA for 64145c4

Commits on May 12, 2026

• Merge pull request #5441 from samanklesaria:fix-transforms-doctest

  Show description for e9aaebe
  

  Flax Authors

  committed
  e9aaebe

  Copy full SHA for e9aaebe

• Merge pull request #5440 from samanklesaria:fix-graphlib-doctest

  Show description for d07c450
  

  Flax Authors

  committed
  d07c450

  Copy full SHA for d07c450

• Merge pull request #5442 from samanklesaria:fix-rnglib-doctest

  Show description for 2ee10c2
  

  Flax Authors

  committed
  2ee10c2

  Copy full SHA for 2ee10c2

• Merge pull request #5425 from samanklesaria:tensorboard_example

  Show description for a9057b8
  

  Flax Authors

  committed
  a9057b8

  Copy full SHA for a9057b8

Commits on May 11, 2026

• add nnx.EMA

  Show description for e9717a1
  

  Cristian Garcia

  authored and

  Flax Authors

  committed
  e9717a1

  Copy full SHA for e9717a1

• Merge pull request #5434 from google:mha-out_sharding

  Show description for 2fa7e66
  

  Flax Authors

  committed
  2fa7e66

  Copy full SHA for 2fa7e66

Commits on May 4, 2026

• Merge branch 'main' into mha-out_sharding

  vfdev-5

  authored
  502012b

  Copy full SHA for 502012b

Commits on Apr 30, 2026

• Merge pull request #5446 from samanklesaria:msgpack_restore_fix

  Show description for 2fe9c86
  

  Flax Authors

  committed
  2fe9c86

  Copy full SHA for 2fe9c86

• Fix doctests in transforms for graph_updates=False default

  Show description for bd6bf36
  samanklesaria

  committed
  bd6bf36

  Copy full SHA for bd6bf36

• Update docstring for split_rngs

  samanklesaria

  committed
  9a36041

  Copy full SHA for 9a36041

Commits on Apr 29, 2026

• Merge pull request #5405 from samanklesaria:minigpt

  Show description for 56b4f38
  

  Flax Authors

  committed
  56b4f38

  Copy full SHA for 56b4f38

• Merge pull request #5443 from samanklesaria:fix-docstring-graphilib

  Show description for 35c5ac5
  

  Flax Authors

  committed
  35c5ac5

  Copy full SHA for 35c5ac5

Commits on Apr 28, 2026

• Merge pull request #5101 from google:rewrite-prefetch_to_device

  Show description for c781023
  

  Flax Authors

  committed
  c781023

  Copy full SHA for c781023

• improve Optimizer

  Show description for 63c1960
  

  Cristian Garcia

  authored and

  Flax Authors

  committed
  63c1960

  Copy full SHA for 63c1960

• fix Pytree typing issue

  Show description for 36d28ae
  

  Cristian Garcia

  authored and

  Flax Authors

  committed
  36d28ae

  Copy full SHA for 36d28ae