Commits

Branch selector

 main

User selector

All users

All time

Commit History

Commits on Oct 25, 2024

• Merge pull request #4337 from 8bitmp3:fixup-haiku-to-flax

  Show description for 0f044c8
  

  Flax Authors

  committed
  0f044c8

  Copy full SHA for 0f044c8

• Fix typos in Flax NNX Migrating from Haiku to Flax

  8bitmp3

  committed
  b83dce7

  Copy full SHA for b83dce7

• Merge pull request #4332 from jakevdp:fix-lift-test

  Show description for cd5db46
  

  Flax Authors

  committed
  cd5db46

  Copy full SHA for cd5db46

• Remove remaining references to Penzai from NNX.

  Show description for 3895d51
  

  danieldjohnson

  authored and

  Flax Authors

  committed
  3895d51

  Copy full SHA for 3895d51

Commits on Oct 24, 2024

• Avoid assert_array_equal on PRNG keys.

  Show description for 27c06b6
  jakevdp

  committed
  27c06b6

  Copy full SHA for 27c06b6

• Avoid using float32 in normalization for mean/var and scale/bias parameters when force_float32_reductions=False

  Show description for 40f080e
  

  eunjaekim-0

  authored and

  Flax Authors

  committed
  40f080e

  Copy full SHA for 40f080e

• Remove GeGLU activation function and golden tests.

  Show description for ad90642
  

  levskaya

  authored and

  Flax Authors

  committed
  ad90642

  Copy full SHA for ad90642

• Merge pull request #4331 from levskaya:fix_lint

  Show description for 85ca2c0
  

  Flax Authors

  committed
  85ca2c0

  Copy full SHA for 85ca2c0

• Merge pull request #4324 from jakevdp:rm-compat

  Show description for 1e88500
  

  Flax Authors

  committed
  1e88500

  Copy full SHA for 1e88500

• fix ruff complaints

  levskaya

  committed
  8be82b4

  Copy full SHA for 8be82b4

• Avoid depending on JAX internals, which are about to change.

  Show description for c12ba53
  

  dougalm

  authored and

  Flax Authors

  committed
  c12ba53

  Copy full SHA for c12ba53

Commits on Oct 23, 2024

• Remove outdated compatibility code.

  Show description for 81efeef
  jakevdp

  committed
  81efeef

  Copy full SHA for 81efeef

• Merge pull request #4322 from google:nnx-cleanup-nnx-basics

  Show description for 4b4b19f
  

  Flax Authors

  committed
  4b4b19f

  Copy full SHA for 4b4b19f

• Merge pull request #4291 from google:init-cache-docs

  Show description for 3f3c03b
  

  Flax Authors

  committed
  3f3c03b

  Copy full SHA for 3f3c03b

• Delete the `nnx.bridge.module` code because we no longer go down that third module class path.

  Show description for 3d9c7e0
  

  IvyZX

  authored and

  Flax Authors

  committed
  3d9c7e0

  Copy full SHA for 3d9c7e0

• remove markdown from section titles

  cgarciae

  committed
  52c07bf

  Copy full SHA for 52c07bf

Commits on Oct 22, 2024

• Merge pull request #4075 from google:nnx-gemma-notebook

  Show description for dc3017a
  

  Flax Authors

  committed
  dc3017a

  Copy full SHA for dc3017a

• Merge pull request #4313 from jorisSchaller:fix_documentation

  Show description for 72284c5
  

  Flax Authors

  committed
  72284c5

  Copy full SHA for 72284c5

• Merge pull request #4282 from vfdev-5:fix-nnx-examples-bad-links

  Show description for ff12eae
  

  Flax Authors

  committed
  ff12eae

  Copy full SHA for ff12eae

• [nnx] add gemma notebook

  cgarciae

  committed
  37a224e

  Copy full SHA for 37a224e

• Merge pull request #4317 from google:improve-state-loading

  Show description for 8360b7c
  

  Flax Authors

  committed
  8360b7c

  Copy full SHA for 8360b7c

Commits on Oct 21, 2024

• Merge pull request #4289 from 8bitmp3:update-evolution-linen-to-nn

  Show description for 092ba1d
  

  Flax Authors

  committed
  092ba1d

  Copy full SHA for 092ba1d

• Merge pull request #4316 from google:improve-mnist-tutorial-2

  Show description for 685b255
  

  Flax Authors

  committed
  685b255

  Copy full SHA for 685b255

• [nnx] improve mnist tutorial

  cgarciae

  committed
  d2606ce

  Copy full SHA for d2606ce

• [nnx] try casting integers keys in State.replace_by_pure_dict

  cgarciae

  committed
  059743c

  Copy full SHA for 059743c

Commits on Oct 20, 2024

• Fix philosophy link

  jorisSchaller

  committed
  df85249

  Copy full SHA for df85249

Commits on Oct 18, 2024

• Merge pull request #4305 from IvyZX:haiku-nnx

  Show description for c1046ad
  

  Flax Authors

  committed
  c1046ad

  Copy full SHA for c1046ad

• Avoid float32 promotion in BatchNorm when using_running_average=True and force_float32_reductions=False

  Show description for a7f32a4
  

  eunjaekim-0

  authored and

  Flax Authors

  committed
  a7f32a4

  Copy full SHA for a7f32a4

Commits on Oct 17, 2024

• Update Flax Evolution from Linen to NNX guide

  8bitmp3

  committed
  f00917b

  Copy full SHA for f00917b

• Merge pull request #4302 from 8bitmp3:add-graphdef-docstring

  Show description for 702afbe
  

  Flax Authors

  committed
  702afbe

  Copy full SHA for 702afbe

• Flesh out the Haiku/Flax guide

  IvyZX

  committed
  ba63f4c

  Copy full SHA for ba63f4c

Commits on Oct 16, 2024

• Merge pull request #4292 from google:v0.10.0

  Show description for c692114
  

  Flax Authors

  committed
  c692114

  Copy full SHA for c692114

• Merge pull request #4284 from 8bitmp3:lint-glossary

  Show description for 168aa3b
  

  Flax Authors

  committed
  168aa3b

  Copy full SHA for 168aa3b

• Add Flax NNX GraphDef docstring

  8bitmp3

  committed
  74423bc

  Copy full SHA for 74423bc

• Merge pull request #4272 from zinccat:rnn_nnx

  Show description for 3bf732c
  

  Flax Authors

  committed
  3bf732c

  Copy full SHA for 3bf732c