Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Mar 30, 2024

• Update community_examples.rst

  mashaan14

  authored
  e2210ad

  Copy full SHA for e2210ad

• Update community_examples.rst

  mashaan14

  authored
  9a595e6

  Copy full SHA for 9a595e6

Commits on Mar 29, 2024

• Merge pull request #3796 from chiamp:graphnode

  Show description for 514c111
  

  Flax Authors

  committed
  514c111

  Copy full SHA for 514c111

• expose nnx.GraphNode

  chiamp

  committed
  1eb988d

  Copy full SHA for 1eb988d

Commits on Mar 28, 2024

• Exposing the experimental _split_transpose JAX scan parameter in Flax.

  Show description for 4671ddd
  

  Flax Team

  committed
  4671ddd

  Copy full SHA for 4671ddd

• Merge pull request #3783 from google:nnx-jit-accepts-many-Modules

  Show description for fe99574
  

  Flax Authors

  committed
  fe99574

  Copy full SHA for fe99574

• [nnx] jit accepts many Modules

  cgarciae

  committed
  e452f5a

  Copy full SHA for e452f5a

Commits on Mar 27, 2024

• Merge pull request #3790 from google:nnx-graph-node-base

  Show description for 6f1f1ef
  

  Flax Authors

  committed
  6f1f1ef

  Copy full SHA for 6f1f1ef

• [nnx] add GraphNode base class

  cgarciae

  committed
  6948138

  Copy full SHA for 6948138

• Merge pull request #3791 from google:nnx-arrays-are-state

  Show description for 0ab0365
  

  Flax Authors

  committed
  0ab0365

  Copy full SHA for 0ab0365

• [nnx] Arrays are state

  cgarciae

  committed
  8718110

  Copy full SHA for 8718110

Commits on Mar 26, 2024

• Merge pull request #3785 from chiamp:pytreenode

  Show description for b57e3aa
  

  Flax Authors

  committed
  b57e3aa

  Copy full SHA for b57e3aa

Commits on Mar 22, 2024

• extended pytreenode to take dataclass kwargs

  chiamp

  committed
  9ccd27d

  Copy full SHA for 9ccd27d

• Merge pull request #3651 from chiamp:kw_only

  Show description for 135b652
  

  Flax Authors

  committed
  135b652

  Copy full SHA for 135b652

• Merge pull request #3772 from chiamp:tabulate_norm

  Show description for c53a515
  

  Flax Authors

  committed
  c53a515

  Copy full SHA for c53a515

• added kw_only dataclass test

  chiamp

  committed
  78e8056

  Copy full SHA for 78e8056

• Use short-circuiting access to debug_key_reuse

  Show description for fb3107f
  

  Jake VanderPlas

  authored and

  Flax Authors

  committed
  fb3107f

  Copy full SHA for fb3107f

Commits on Mar 21, 2024

• Account for JAX flag name change in https://github.com/google/jax/pull/20370

  Show description for f234fc1
  

  Jake VanderPlas

  authored and

  Flax Authors

  committed
  f234fc1

  Copy full SHA for f234fc1

• Merge pull request #3768 from google:nnx-pytree-are-trees

  Show description for 12b919c
  

  Flax Authors

  committed
  12b919c

  Copy full SHA for 12b919c

Commits on Mar 20, 2024

• Merge pull request #3773 from chiamp:mgu_test

  Show description for 20213c9
  

  Flax Authors

  committed
  20213c9

  Copy full SHA for 20213c9

• Merge pull request #3588 from canyon289:doc_improvements

  Show description for d75c2e5
  

  Flax Authors

  committed
  d75c2e5

  Copy full SHA for d75c2e5

• Access thread_resources via jax.interpreters.pxla instead of jax.experimental.maps

  Show description for e1465fe
  

  superbobry

  authored and

  Flax Authors

  committed
  e1465fe

  Copy full SHA for e1465fe

• added MGU reset_gate test

  chiamp

  committed
  cfb4d9d

  Copy full SHA for cfb4d9d

• fix tabulate on norm wrappers

  chiamp

  committed
  4f561be

  Copy full SHA for 4f561be

Commits on Mar 19, 2024

• Merge pull request #3760 from carlosgmartin:reset_gate_flag

  Show description for 006da41
  

  Flax Authors

  committed
  006da41

  Copy full SHA for 006da41

• Add reset_gate flag to MGUCell.

  carlosgmartin

  committed
  f932fdb

  Copy full SHA for f932fdb

• [nnx] pytree are trees

  cgarciae

  committed
  7317d9b

  Copy full SHA for 7317d9b

• Merge pull request #3769 from chiamp:note

  Show description for 8220154
  

  Flax Authors

  committed
  8220154

  Copy full SHA for 8220154

• Merge pull request #3759 from google:nnx-improve-graph-update

  Show description for b626099
  

  Flax Authors

  committed
  b626099

  Copy full SHA for b626099

• [nnx] add ref_cache to graph_unflatten

  cgarciae

  committed
  c8cf3f5

  Copy full SHA for c8cf3f5

• use note box in docstrings

  chiamp

  committed
  f14d8cc

  Copy full SHA for f14d8cc

• Merge pull request #3767 from google:improve-make_rng-docstring

  Show description for 79851d4
  

  Flax Authors

  committed
  79851d4

  Copy full SHA for 79851d4

• Merge pull request #3765 from chiamp:docstring_init_apply

  Show description for 2f76a92
  

  Flax Authors

  committed
  2f76a92

  Copy full SHA for 2f76a92

Commits on Mar 18, 2024

• updated rng docstring for init and apply

  chiamp

  committed
  76f6c46

  Copy full SHA for 76f6c46

• Merge pull request #3762 from chiamp:nnx_api_reference

  Show description for a6e771f
  

  Flax Authors

  committed
  a6e771f

  Copy full SHA for a6e771f

Pagination

Previous

Next