google/flax

Pull Requests Commits

Merge pull request #4333 from google:improve-transforms-guide

a-googler committed 1 year ago

58d1e537

[nnx] cleanup gemma notebook

cgarciae committed 1 year ago

158bcea2

[nnx] improve transforms guide

cgarciae committed 1 year ago

a8e2ab1f

Merge branch 'main' into upgrade-haiku-linen-to-nnx

8bitmp3 committed 1 year ago

Verified 84fa3167

Upgrade Flax NNX Migrating from Haiku doc

8bitmp3 committed 1 year ago

4ed2ba8c

Merge pull request #4338 from IvyZX:docfix

a-googler committed 1 year ago

ab122af7

Add API reference for flax.nnx.nn and improve landing page

IvyZX committed 1 year ago

e1415041

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

a-googler committed 1 year ago

0f044c8b

Fix typos in Flax NNX Migrating from Haiku to Flax

8bitmp3 committed 1 year ago

b83dce71

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

a-googler committed 1 year ago

cd5db467

Remove remaining references to Penzai from NNX.

danieldjohnson committed 1 year ago

3895d51b

Avoid assert_array_equal on PRNG keys.

jakevdp committed 1 year ago

27c06b62

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

eunjaekim-0 committed 1 year ago

40f080ed

Remove GeGLU activation function and golden tests.

levskaya committed 1 year ago

ad90642b

Merge pull request #4331 from levskaya:fix_lint

a-googler committed 1 year ago

85ca2c0e

Merge pull request #4324 from jakevdp:rm-compat

a-googler committed 1 year ago

1e885003

fix ruff complaints

levskaya committed 1 year ago

8be82b44

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

dougalm committed 1 year ago

c12ba53e

Remove outdated compatibility code.

jakevdp committed 1 year ago

81efeef0

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

a-googler committed 1 year ago

4b4b19f0

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

a-googler committed 1 year ago

3f3c03b2

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

IvyZX committed 1 year ago

3d9c7e03

remove markdown from section titles

cgarciae committed 1 year ago

52c07bf2

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

a-googler committed 1 year ago

dc3017aa

Merge pull request #4313 from jorisSchaller:fix_documentation

a-googler committed 1 year ago

72284c5e

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

a-googler committed 1 year ago

ff12eae7

[nnx] add gemma notebook

cgarciae committed 1 year ago

37a224e9

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

a-googler committed 1 year ago

8360b7c2

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

a-googler committed 1 year ago

092ba1d2

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

a-googler committed 1 year ago

685b255f

Newer Older