Commits

Breadcrumbs

History for

onmain

User selector

All users

All time

Commit History

Commits on Apr 23, 2026

• 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

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

• Avoid deprecated JAX API

  Show description for 7ee4043
  

  Jake VanderPlas

  authored and

  Flax Authors

  committed
  7ee4043

  Copy full SHA for 7ee4043

Commits on Apr 13, 2026

• Merge pull request #5388 from samanklesaria:rename_abstract

  Show description for d516e45
  

  Flax Authors

  committed
  d516e45

  Copy full SHA for d516e45

• Merge pull request #5379 from mohsinm-dev:fix-rngs-dunder-getattr

  Show description for 228c50e
  

  Flax Authors

  committed
  228c50e

  Copy full SHA for 228c50e

• Merge pull request #5381 from samanklesaria:variables-in-merge

  Show description for 4c9d11f
  

  Flax Authors

  committed
  4c9d11f

  Copy full SHA for 4c9d11f

• Merge pull request #5377 from google:fix-sow-link

  Show description for 4e73018
  

  Flax Authors

  committed
  4e73018

  Copy full SHA for 4e73018

Commits on Apr 10, 2026

• improve nnx.compat

  Show description for 4fd69fa
  

  Cristian Garcia

  authored and

  Flax Authors

  committed
  4fd69fa

  Copy full SHA for 4fd69fa

• forward fix simple transforms by default

  Show description for b5a85d2
  

  Cristian Garcia

  authored and

  Flax Authors

  committed
  b5a85d2

  Copy full SHA for b5a85d2

• Fix for change to `jax.Array.__contains__`

  Show description for 64aa65a
  

  Jake VanderPlas

  authored and

  Flax Authors

  committed
  64aa65a

  Copy full SHA for 64aa65a

Commits on Apr 7, 2026

• fix ci

  cgarciae

  committed
  a3bd2ca

  Copy full SHA for a3bd2ca

• change defaults for Dropout and BatchNorm

  Show description for 84da828
  

  Cristian Garcia

  authored and

  Flax Authors

  committed
  84da828

  Copy full SHA for 84da828

• Merge pull request #5389 from samanklesaria:rename_pure

  Show description for d874278
  

  Flax Authors

  committed
  d874278

  Copy full SHA for d874278

• fix Rngs.__getattr__ returning default stream for dunder names (#5375)

  

  mohsinm-dev

  authored and

  samanklesaria

  committed
  4f6b33f

  Copy full SHA for 4f6b33f

• Rename to 'recreate_variables'

  samanklesaria

  committed
  5716ff7

  Copy full SHA for 5716ff7

• Add flag for variables in nnx.merge

  samanklesaria

  committed
  148f202

  Copy full SHA for 148f202

• Fix the wrong names for the hooks in nnx.with_metadata, fixed such that names match the proper hook names in Variable class.

  Show description for 5a4859d
  

  Flax Team

  committed
  5a4859d

  Copy full SHA for 5a4859d

• [pmap] In-line definitions of `jax.device_put_sharded` and `jax.device_put_replicated`.

  Show description for 355e141
  

  danielsuo

  authored and

  Flax Authors

  committed
  355e141

  Copy full SHA for 355e141

Commits on Apr 2, 2026

• improve with_rngs

  Show description for 983506b
  

  Cristian Garcia

  authored and

  Flax Authors

  committed
  983506b

  Copy full SHA for 983506b

• add Rngs.broadcast

  Show description for 3645777
  

  Cristian Garcia

  authored and

  Flax Authors

  committed
  3645777

  Copy full SHA for 3645777

• add nnx.prefix

  Show description for cd61953
  

  Cristian Garcia

  authored and

  Flax Authors

  committed
  cd61953

  Copy full SHA for cd61953

• Rename pure -> as_pure

  samanklesaria

  committed
  6132155

  Copy full SHA for 6132155

Commits on Apr 1, 2026

• Deprecate the keep_rngs argument in recurrent.py

  Show description for 3c68194
  

  Flax Team

  committed
  3c68194

  Copy full SHA for 3c68194