Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Dec 5, 2024

• Update references to JAX's GitHub repo

  Show description for 3e789b7
  

  jakeharmon8

  authored and

  Flax Authors

  committed
  3e789b7

  Copy full SHA for 3e789b7

Commits on Dec 4, 2024

• Merge pull request google#4385 from emmanuel-ferdman:main

  Show description for 30b438c
  

  Flax Authors

  committed
  30b438c

  Copy full SHA for 30b438c

• Merge pull request google#4410 from google:nnx-optimizations-p3

  Show description for 7eac051
  

  Flax Authors

  committed
  7eac051

  Copy full SHA for 7eac051

• [nnx] optimize graph

  cgarciae

  committed
  217f5ba

  Copy full SHA for 217f5ba

• Merge pull request google#4415 from tilakrayal:patch-3

  Show description for 6c84097
  

  Flax Authors

  committed
  6c84097

  Copy full SHA for 6c84097

• Merge pull request google#4416 from rajasekharporeddy:main

  Show description for 27b1829
  

  Flax Authors

  committed
  27b1829

  Copy full SHA for 27b1829

• Remove the repeated methods in flax.nnx.Module documentation

  rajasekharporeddy

  committed
  63a1b7d

  Copy full SHA for 63a1b7d

• Update linen_to_nnx.rst

  tilakrayal

  authored
  4756b34

  Copy full SHA for 4756b34

Commits on Dec 1, 2024

• Merge pull request google#4279 from 8bitmp3:update-nnx-randomness

  Show description for a5e62af
  

  Flax Authors

  committed
  a5e62af

  Copy full SHA for a5e62af

Commits on Nov 29, 2024

• Upgrade Flax NNX Randomness

  8bitmp3

  committed
  f16e6d6

  Copy full SHA for f16e6d6

Commits on Nov 27, 2024

• Merge pull request google#4400 from google:nnx-more-optimizations-p2

  Show description for 5d896bc
  

  Flax Authors

  committed
  5d896bc

  Copy full SHA for 5d896bc

• Merge pull request google#4402 from tilakrayal:patch-2

  Show description for be2f861
  

  Flax Authors

  committed
  be2f861

  Copy full SHA for be2f861

• [nnx] optimize Variable

  cgarciae

  committed
  7b50ffe

  Copy full SHA for 7b50ffe

• Update haiku_to_flax.rst

  tilakrayal

  authored
  18b59dd

  Copy full SHA for 18b59dd

Commits on Nov 26, 2024

• Merge pull request google#4399 from google:nnx-more-optimizations

  Show description for fe8fb59
  

  Flax Authors

  committed
  fe8fb59

  Copy full SHA for fe8fb59

• [nnx] optimize NodeDef.attributes

  cgarciae

  committed
  861936f

  Copy full SHA for 861936f

• Automated Code Change: Suppress pytype errors

  Show description for 6bc9858
  

  Martin Huschenbett

  authored and

  Flax Authors

  committed
  6bc9858

  Copy full SHA for 6bc9858

• Merge pull request google#4384 from google:nnx-performance-guide

  Show description for abc1155
  

  Flax Authors

  committed
  abc1155

  Copy full SHA for abc1155

Commits on Nov 25, 2024

• [nnx] add performance guide notebook

  cgarciae

  committed
  94793f9

  Copy full SHA for 94793f9

Commits on Nov 20, 2024

• Merge pull request google#4386 from 8bitmp3:update-flax-readme

  Show description for be26138
  

  Flax Authors

  committed
  be26138

  Copy full SHA for be26138

• Upgrade Flax readme to NNX

  8bitmp3

  committed
  bddf5f2

  Copy full SHA for bddf5f2

• Merge pull request google#4390 from IvyZX:loop

  Show description for 4e25898
  

  Flax Authors

  committed
  4e25898

  Copy full SHA for 4e25898

Commits on Nov 19, 2024

• Fix fori_loop and while_loop on multiple modules

  IvyZX

  committed
  19a4abf

  Copy full SHA for 19a4abf

• Merge pull request google#4387 from google:v0.10.2

  Show description for d89c955
  

  Flax Authors

  committed
  d89c955

  Copy full SHA for d89c955

Commits on Nov 18, 2024

• v0.10.2

  cgarciae

  committed
  52a8c48

  Copy full SHA for 52a8c48

Commits on Nov 16, 2024

• Update async_checkpointer.py reference

  Show description for 6a4a957
  emmanuel-ferdman

  committed
  6a4a957

  Copy full SHA for 6a4a957

• Merge pull request google#4377 from google:nnx-pytree-optimization

  Show description for 6382a2b
  

  Flax Authors

  committed
  6382a2b

  Copy full SHA for 6382a2b

• [nnx] add some optimizations to graph.py

  cgarciae

  committed
  ba57524

  Copy full SHA for ba57524

• [nnx] use HashableMapping instead of FrozenDict

  cgarciae

  committed
  7fd97dc

  Copy full SHA for 7fd97dc

Commits on Nov 15, 2024

• Merge pull request google#4370 from 8bitmp3:update-nnx-fori_loop-docstring

  Show description for 91bf758
  

  Flax Authors

  committed
  91bf758

  Copy full SHA for 91bf758

• Merge pull request google#4371 from 8bitmp3:update-nnx-while_loop

  Show description for 9147a7c
  

  Flax Authors

  committed
  9147a7c

  Copy full SHA for 9147a7c

• Merge pull request google#4381 from google:nnx-checkify

  Show description for d13f047
  

  Flax Authors

  committed
  d13f047

  Copy full SHA for d13f047

• [nnx] add checkify

  cgarciae

  committed
  ec73cfc

  Copy full SHA for ec73cfc

Commits on Nov 14, 2024

• Merge pull request google#4373 from 8bitmp3:add-nnx-remat-docstring

  Show description for 856b6b5
  

  Flax Authors

  committed
  856b6b5

  Copy full SHA for 856b6b5

• Add flax.nnx.remat docstring

  8bitmp3

  committed
  8fbe99c

  Copy full SHA for 8fbe99c

Pagination

Previous

Next