Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Oct 10, 2025

• Merge pull request #5016 from google:gym-cleanup

  Show description for f6ed5ec
  

  Flax Authors

  committed
  f6ed5ec

  Copy full SHA for f6ed5ec

• Merge pull request #5006 from google:variable-refactor

  Show description for 1d3a2ee
  

  Flax Authors

  committed
  1d3a2ee

  Copy full SHA for 1d3a2ee

• Variable refactor

  cgarciae

  committed
  33fbbe6

  Copy full SHA for 33fbbe6

• Merge pull request #5015 from google:optimize-scan-transpose

  Show description for 59cfeed
  

  Flax Authors

  committed
  59cfeed

  Copy full SHA for 59cfeed

• Merge pull request #5017 from google:update-smpd-guide

  Show description for b38601d
  

  Flax Authors

  committed
  b38601d

  Copy full SHA for b38601d

• add Rngs KeylessInitializers

  cgarciae

  committed
  d96c9ed

  Copy full SHA for d96c9ed

Commits on Oct 9, 2025

• Migrate from `jax.experimental.enable_x64` to `jax.enable_x64`.

  Show description for 1453d29
  

  Jake VanderPlas

  authored and

  Flax Authors

  committed
  1453d29

  Copy full SHA for 1453d29

• Merge pull request #4998 from google:remove-kernel-init-from-layers

  Show description for bb51943
  

  Flax Authors

  committed
  bb51943

  Copy full SHA for bb51943

• Merge pull request #5009 from jburnim:jburnim_sphinx_upgrade

  Show description for e8e8435
  

  Flax Authors

  committed
  e8e8435

  Copy full SHA for e8e8435

• Remove invalid gymnasium dependency in pyproject.toml

  IvyZX

  authored
  492f063

  Copy full SHA for 492f063

• optimize scan transpose

  cgarciae

  committed
  4727490

  Copy full SHA for 4727490

Commits on Oct 8, 2025

• Use sphinx 6.2+ for docs, which works with Python 3.13.

  jburnim

  committed
  00097aa

  Copy full SHA for 00097aa

Commits on Oct 7, 2025

• Merge pull request #4993 from google:no-setattr-metadata-p2

  Show description for 75fd8fa
  

  Flax Authors

  committed
  75fd8fa

  Copy full SHA for 75fd8fa

• Merge pull request #5001 from jburnim:jburnim_python3.13

  Show description for f2942d7
  

  Flax Authors

  committed
  f2942d7

  Copy full SHA for f2942d7

Commits on Oct 6, 2025

• Avoid install, import, or tests with tensorflow-text under Python 3.13+.

  jburnim

  committed
  0a0645c

  Copy full SHA for 0a0645c

• Updated recurrent layers

  vfdev-5

  committed
  e1caa82

  Copy full SHA for e1caa82

Commits on Oct 4, 2025

• Removed *_init atttributes from layers

  vfdev-5

  committed
  d81ce1d

  Copy full SHA for d81ce1d

Commits on Oct 3, 2025

• Merge pull request #4961 from samanklesaria:issues/4943

  Show description for 771eadb
  

  Flax Authors

  committed
  771eadb

  Copy full SHA for 771eadb

Commits on Oct 1, 2025

• Merge pull request #4985 from google:no-setattr-metadata

  Show description for 6080725
  

  Flax Authors

  committed
  6080725

  Copy full SHA for 6080725

• disallow setting metadata through settattr

  cgarciae

  committed
  e30c934

  Copy full SHA for e30c934

• improve set/get_metadata

  cgarciae

  committed
  c23304d

  Copy full SHA for c23304d

• Merge pull request #4623 from mattbahr:implement-spectral-and-instance-norm

  Show description for 8025b3d
  

  Flax Authors

  committed
  8025b3d

  Copy full SHA for 8025b3d

• Update flax/nnx/nn/normalization.py

  vfdev-5

  authored
  4721a6e

  Copy full SHA for 4721a6e

• Merge pull request #4992 from google:fix-nnx-diff

  Show description for 21d1289
  

  Flax Authors

  committed
  21d1289

  Copy full SHA for 21d1289

• Merge pull request #4991 from google:use-jnp-stack-in-stack_forest

  Show description for 8b8b929
  

  Flax Authors

  committed
  8b8b929

  Copy full SHA for 8b8b929

Commits on Sep 30, 2025

• Fixed broken nnx.statelib.diff

  vfdev-5

  committed
  11a4264

  Copy full SHA for 11a4264

• use jnp.stack instead of np.stack

  vfdev-5

  authored
  c08e60b

  Copy full SHA for c08e60b

Commits on Sep 29, 2025

• Future-proof imports of jax.new_ref / jax.Ref.

  Show description for 54fa32d
  

  Jake VanderPlas

  authored and

  Flax Authors

  committed
  54fa32d

  Copy full SHA for 54fa32d

• Improved nnx SpectralNorm implementation

  vfdev-5

  committed
  77a8c4c

  Copy full SHA for 77a8c4c

• Merge pull request #4891 from mohsinm-dev:fix-nnx-tabulate-empty-dict-issue-4889

  Show description for f65afdb
  

  Flax Authors

  committed
  f65afdb

  Copy full SHA for f65afdb

• Merge pull request #4948 from samanklesaria:issues/4883

  Show description for ebb980a
  

  Flax Authors

  committed
  ebb980a

  Copy full SHA for ebb980a

• clean up jax.Ref-related names

  Show description for 4c4bc3b
  

  Cristian Garcia

  authored and

  Flax Authors

  committed
  4c4bc3b

  Copy full SHA for 4c4bc3b

Commits on Sep 26, 2025

• Merge pull request #4980 from IvyZX:shardref

  Show description for a9dccc0
  

  Flax Authors

  committed
  a9dccc0

  Copy full SHA for a9dccc0

• Clarify `*Norm` layer docstrings: `axis_index_groups` is unused under SPMD jit.

  Show description for ae1594b
  

  arnoegw

  authored and

  Flax Authors

  committed
  ae1594b

  Copy full SHA for ae1594b

• Remove type annotation

  samanklesaria

  committed
  8b7c8bd

  Copy full SHA for 8b7c8bd

Pagination

Previous

Next