Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Dec 5, 2025

• [flax:examples:nlp_seq] Create a main.py file to run tests with config files to match other examples. #jax-fixit

  Show description for 872d50b
  

  danielsuo

  authored and

  Flax Authors

  committed
  872d50b

  Copy full SHA for 872d50b

• [flax:examples:lm1b_nnx] Update example to work internally. #jax-fixit.

  Show description for 881b8ed
  

  danielsuo

  authored and

  Flax Authors

  committed
  881b8ed

  Copy full SHA for 881b8ed

Commits on Dec 4, 2025

• Avoid passing `concrete` argument to `jax.remat`

  Show description for ec85cdd
  

  Jake VanderPlas

  authored and

  Flax Authors

  committed
  ec85cdd

  Copy full SHA for ec85cdd

Commits on Dec 3, 2025

• [flax:examples:ppo] Fix some linter / import issues. #jax-fixit

  Show description for 59cfd99
  

  danielsuo

  authored and

  Flax Authors

  committed
  59cfd99

  Copy full SHA for 59cfd99

Commits on Dec 2, 2025

• [flax:examples:sst2] Fix notebook error.

  Show description for 5a3d148
  

  danielsuo

  authored and

  Flax Authors

  committed
  5a3d148

  Copy full SHA for 5a3d148

• Merge pull request #5094 from thijs-vanweezel:filters

  Show description for a5af24c
  

  Flax Authors

  committed
  a5af24c

  Copy full SHA for a5af24c

Commits on Dec 1, 2025

• [flax:examples:sst2] Fix pytype errors.

  Show description for 6ac5a78
  

  danielsuo

  authored and

  Flax Authors

  committed
  6ac5a78

  Copy full SHA for 6ac5a78

• [flax:examples:gemma] Fixing linter errors.

  Show description for 324f4b4
  

  danielsuo

  authored and

  Flax Authors

  committed
  324f4b4

  Copy full SHA for 324f4b4

• [flax:examples:vae] Small linter fixes.

  Show description for 88d7497
  

  danielsuo

  authored and

  Flax Authors

  committed
  88d7497

  Copy full SHA for 88d7497

• [flax:examples:seq2seq] Create main and default config based on seq2seq.ipynb.

  Show description for 9ed59e2
  

  danielsuo

  authored and

  Flax Authors

  committed
  9ed59e2

  Copy full SHA for 9ed59e2

• [flax:examples:wmt] Small linter fixes.

  Show description for c29b6d6
  

  danielsuo

  authored and

  Flax Authors

  committed
  c29b6d6

  Copy full SHA for c29b6d6

Commits on Nov 19, 2025

• Merge pull request #5095 from google:jax-minver-0.8.1

  Show description for 697f4e5
  

  Flax Authors

  committed
  697f4e5

  Copy full SHA for 697f4e5

• update jax minver to 0.8.1

  cgarciae

  committed
  b424c49

  Copy full SHA for b424c49

• hint that key is cast to string

  thijs-vanweezel

  committed
  a7bb109

  Copy full SHA for a7bb109

• test pathcontains with exact=false

  thijs-vanweezel

  committed
  12b2e42

  Copy full SHA for 12b2e42

• allow substring matching

  thijs-vanweezel

  committed
  3d34055

  Copy full SHA for 3d34055

Commits on Nov 18, 2025

• Merge pull request #5097 from google:fix-main-4

  Show description for bee81ed
  

  Flax Authors

  committed
  bee81ed

  Copy full SHA for bee81ed

• Fixed deprecated .value usage failing CI tests

  vfdev-5

  committed
  00cf44e

  Copy full SHA for 00cf44e

Commits on Nov 17, 2025

• Merge pull request #5084 from samanklesaria:variable_eq

  Show description for 09537f4
  

  Flax Authors

  committed
  09537f4

  Copy full SHA for 09537f4

• Merge pull request #5052 from google:deprecate-value-2

  Show description for b7e66b7
  

  Flax Authors

  committed
  b7e66b7

  Copy full SHA for b7e66b7

• deprecate .value

  cgarciae

  committed
  a2eac32

  Copy full SHA for a2eac32

• Merge pull request #5055 from mohsinm-dev:fix-bound-method-auto-unbinding

  Show description for d6e8e14
  

  Flax Authors

  committed
  d6e8e14

  Copy full SHA for d6e8e14

• Merge pull request #5073 from google:fix-nnx-scan-carry-pytree-sow

  Show description for 544a050
  

  Flax Authors

  committed
  544a050

  Copy full SHA for 544a050

• Merge pull request #5071 from chapman20j:set_mode_info

  Show description for 0b1d841
  

  Flax Authors

  committed
  0b1d841

  Copy full SHA for 0b1d841

• Merge branch 'main' into pr/5055

  samanklesaria

  committed
  dece6b3

  Copy full SHA for dece6b3

Commits on Nov 14, 2025

• Merge pull request #5090 from google:fix-main-3

  Show description for 5109e2c
  

  Flax Authors

  committed
  5109e2c

  Copy full SHA for 5109e2c

• fix main

  cgarciae

  committed
  d982784

  Copy full SHA for d982784

• backfill eager_sharding in set_metadata

  Show description for 1ac99b0
  

  Cristian Garcia

  authored and

  Flax Authors

  committed
  1ac99b0

  Copy full SHA for 1ac99b0

• add HijaxVariable [rollforward]

  Show description for b5db513
  

  Cristian Garcia

  authored and

  Flax Authors

  committed
  b5db513

  Copy full SHA for b5db513

Commits on Nov 13, 2025

• Add __eq__ for variables

  samanklesaria

  committed
  1d2a672

  Copy full SHA for 1d2a672

Commits on Nov 12, 2025

• Merge pull request #5079 from samanklesaria:eager_sharding_context

  Show description for 7710c30
  

  Flax Authors

  committed
  7710c30

  Copy full SHA for 7710c30

• Merge pull request #5074 from google:params_metadata_args

  Show description for b08cb20
  

  Flax Authors

  committed
  b08cb20

  Copy full SHA for b08cb20

• replaced carry_graphdefs -> graphdefs

  vfdev-5

  committed
  aa02f70

  Copy full SHA for aa02f70

• Implement *_metadata to other layers

  vfdev-5

  committed
  f320a5c

  Copy full SHA for f320a5c

• Added kernel_metadata/bias_metadata args to nnx layers

  Show description for c603655
  vfdev-5

  committed
  c603655

  Copy full SHA for c603655

Pagination

Previous

Next