Commits

Branch selector

 main

User selector

All users

All time

Commit History

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

• Merge pull request #5431 from samanklesaria:machine_translation

  Show description for 48141ec
  

  Flax Authors

  committed
  48141ec

  Copy full SHA for 48141ec

• Merge pull request #5384 from samanklesaria:vars_as_rename

  Show description for e3a73c0
  

  Flax Authors

  committed
  e3a73c0

  Copy full SHA for e3a73c0

• Merge pull request #5408 from google:migration-pytorch

  Show description for 9f17346
  

  Flax Authors

  committed
  9f17346

  Copy full SHA for 9f17346

• Rename vars_as to with_vars

  samanklesaria

  committed
  d07091f

  Copy full SHA for d07091f

• Merge pull request #5424 from samanklesaria:example_cleanup

  Show description for 0ccef55
  

  Flax Authors

  committed
  0ccef55

  Copy full SHA for 0ccef55

• Merge pull request #5416 from samanklesaria:view_guide_change

  Show description for b1f0e22
  

  Flax Authors

  committed
  b1f0e22

  Copy full SHA for b1f0e22

• Merge pull request #5409 from google:gemma-example-using-grain--fix-intermediates

  Show description for 8671e60
  

  Flax Authors

  committed
  8671e60

  Copy full SHA for 8671e60

• Updated gemma docs

  vfdev-5

  committed
  b788d79

  Copy full SHA for b788d79

Commits on Apr 22, 2026

• Disallow `P.UNCONSTRAINED` in `out_shardings` in `jit`. This leads to simplifications in how we lower shardings to stableHLO.

  Show description for 9877bea
  

  yashk2810

  authored and

  Flax Authors

  committed
  9877bea

  Copy full SHA for 9877bea

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

• Add deeper explanations to translation example

  samanklesaria

  committed
  44d96f7

  Copy full SHA for 44d96f7

• Add explanation of nnx.view functions

  samanklesaria

  committed
  aab0c99

  Copy full SHA for aab0c99

• Update view guide to mention with_ and as_ prefixes

  samanklesaria

  committed
  4931b8f

  Copy full SHA for 4931b8f

• Reorganize examples

  samanklesaria

  committed
  b7d9388

  Copy full SHA for b7d9388

• Add machine translation example

  samanklesaria

  committed
  2314a04

  Copy full SHA for 2314a04

• "From PyTorch to JAX and Flax" migration guide

  vfdev-5

  committed
  132102f

  Copy full SHA for 132102f

• Addressed comments

  vfdev-5

  authored
  58658ac

  Copy full SHA for 58658ac

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

• Merge pull request #5421 from google:5412-add-basemod-kwargs-to-lora

  Show description for d8ef528
  

  Flax Authors

  committed
  d8ef528

  Copy full SHA for d8ef528

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

• Fixed intermedites capture

  vfdev-5

  committed
  cdfadd6

  Copy full SHA for cdfadd6

• 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 #5403 from samanklesaria:diffusion

  Show description for 9e73be4
  

  Flax Authors

  committed
  9e73be4

  Copy full SHA for 9e73be4