Commits

Branch selector

 main

User selector

All users

All time

Commit History

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

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

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

Commits on Apr 27, 2026

• Add fix for msgpack serialization of nnx.List structures

  samanklesaria

  committed
  b84b69c

  Copy full SHA for b84b69c

Commits on Apr 24, 2026

• Merge pull request #5098 from google:reenable-pytest-examples

  Show description for af461c1
  

  Flax Authors

  committed
  af461c1

  Copy full SHA for af461c1

• Fix docstrings in graphlib under graph=False

  samanklesaria

  committed
  f17dc02

  Copy full SHA for f17dc02

• Fix update doctest for graph_updates=False default

  Show description for 4f9f0ac
  samanklesaria

  committed
  4f9f0ac

  Copy full SHA for 4f9f0ac

Commits on Apr 23, 2026

• Added tests for flax.jax_utils.prefetch_to_device and flax.jax_utils.replicate

  vfdev-5

  committed
  b055949

  Copy full SHA for b055949

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

  Show description for 7075c6e
  

  Flax Authors

  committed
  7075c6e

  Copy full SHA for 7075c6e

• Merge pull request #5431 from samanklesaria:machine_translation

  Show description for 48141ec
  

  Flax Authors

  committed
  48141ec

  Copy full SHA for 48141ec

• Reenabled examples tests

  vfdev-5

  committed
  0fb638c

  Copy full SHA for 0fb638c

• Merge pull request #5384 from samanklesaria:vars_as_rename

  Show description for e3a73c0
  

  Flax Authors

  committed
  e3a73c0

  Copy full SHA for e3a73c0

• Comment out tensorboard magic for sphinx rendering

  samanklesaria

  committed
  83e3c21

  Copy full SHA for 83e3c21

• Add minigpt example

  samanklesaria

  committed
  2b052ed

  Copy full SHA for 2b052ed