Commits

Breadcrumbs

History for

onmain

User selector

All users

All time

Commit History

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 15, 2026

• Merge pull request #5363 from samanklesaria:rng_guide_update

  Show description for d044de8
  

  Flax Authors

  committed
  d044de8

  Copy full SHA for d044de8

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

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

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

• 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

• 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

• Add minigpt example

  samanklesaria

  committed
  2b052ed

  Copy full SHA for 2b052ed

• Update randomness guide to use split

  samanklesaria

  committed
  81824b4

  Copy full SHA for 81824b4

• Rename vars_as to with_vars

  samanklesaria

  committed
  d07091f

  Copy full SHA for d07091f

Commits on Apr 21, 2026

• Merge pull request #5422 from SAY-5:fix-doubled-the-typos

  Show description for 09ed488
  

  Flax Authors

  committed
  09ed488

  Copy full SHA for 09ed488

• Deprecate Variable hooks in favor of custom Variable types.

  Show description for a13993f
  

  Flax Team

  committed
  a13993f

  Copy full SHA for a13993f

• Added out_sharding and qkv_sharding to MHA call method

  Show description for 42edb5b
  vfdev-5

  committed
  42edb5b

  Copy full SHA for 42edb5b

Commits on Apr 19, 2026

• update version to 0.12.7

  Show description for 66e58ed
  

  Cristian Garcia

  authored and

  Flax Authors

  committed
  66e58ed

  Copy full SHA for 66e58ed

Commits on Apr 18, 2026

• Avoid usage of deprecated `jax.core` APIs.

  Show description for f610964
  

  Cristian Garcia

  authored and

  Flax Authors

  committed
  f610964

  Copy full SHA for f610964

Commits on Apr 16, 2026

• fix 'the the' in nnx/nn/linear.py

  Show description for 6427e72
  SAY-5

  committed
  6427e72

  Copy full SHA for 6427e72

• fix 'the the' in nnx/transforms/compilation.py

  Show description for 3079ffd
  SAY-5

  committed
  3079ffd

  Copy full SHA for 3079ffd

• fix 'the the' in linen/linear.py

  Show description for 3cb8fc9
  SAY-5

  committed
  3cb8fc9

  Copy full SHA for 3cb8fc9

• fix 'the the' in rnglib.py

  Show description for 884384a
  SAY-5

  committed
  884384a

  Copy full SHA for 884384a

• fix doubled 'the the' typos across multiple files

  Show description for 14788d5
  SAY-5

  committed
  14788d5

  Copy full SHA for 14788d5

Commits on Apr 15, 2026

• Added kwargs to call to pass kwargs to basemodule in nnx.LoRA

  Show description for 8fa97d0
  vfdev-5

  committed
  8fa97d0

  Copy full SHA for 8fa97d0

• add nnx.unpack

  Show description for 901efc3
  

  Cristian Garcia

  authored and

  Flax Authors

  committed
  901efc3

  Copy full SHA for 901efc3

Commits on Apr 14, 2026

• use State instead of GraphState

  Show description for 09f0f93
  

  Cristian Garcia

  authored and

  Flax Authors

  committed
  09f0f93

  Copy full SHA for 09f0f93

• improve check_prefix

  Show description for 1c15194
  

  Cristian Garcia

  authored and

  Flax Authors

  committed
  1c15194

  Copy full SHA for 1c15194