google/flax

Pull Requests Commits

Variable refactor

cgarciae committed 91 days ago

33fbbe6b

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

a-googler committed 91 days ago

59cfeed5

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

a-googler committed 91 days ago

b38601d1

add Rngs KeylessInitializers

cgarciae committed 91 days ago

d96c9edf

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

Jake VanderPlas committed 91 days ago

1453d295

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

a-googler committed 91 days ago

bb519438

Merge pull request #5009 from jburnim:jburnim_sphinx_upgrade

a-googler committed 91 days ago

e8e84353

Remove invalid gymnasium dependency in pyproject.toml

IvyZX committed 91 days ago

Verified 492f063d

optimize scan transpose

cgarciae committed 91 days ago

4727490f

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

jburnim committed 93 days ago

00097aa2

Fix nnx.tabulate metadata serialization; add deterministic repr fallback and test

mohsinm-dev committed 93 days ago

Verified b103d0be

Fix nnx.tabulate crash with Variable value hooks

mohsinm-dev committed 93 days ago

Verified 5ccc845a

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

a-googler committed 93 days ago

75fd8fa1

Merge pull request #5001 from jburnim:jburnim_python3.13

a-googler committed 93 days ago

f2942d72

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

jburnim committed 94 days ago

0a0645c0

Updated recurrent layers

vfdev-5 committed 95 days ago

e1caa822

Convert linen pytorch guide to nnx

samanklesaria committed 95 days ago

0c06bdc2

Removed *_init atttributes from layers

vfdev-5 committed 97 days ago

d81ce1db

Merge pull request #4961 from samanklesaria:issues/4943

a-googler committed 98 days ago

771eadb1

Update index.rst example to use explicit randomness

samanklesaria committed 99 days ago

e3004690

Add pros and cons of explicit randomness

samanklesaria committed 99 days ago

819ac067

Use fork_rngs with implicit state in randomness guide

samanklesaria committed 99 days ago

1e0b9ea5

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

a-googler committed 99 days ago

6080725f

disallow setting metadata through settattr

cgarciae committed 99 days ago

e30c9347

improve set/get_metadata

cgarciae committed 99 days ago

c23304db

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

a-googler committed 99 days ago

8025b3dd

Update flax/nnx/nn/normalization.py

vfdev-5 committed 100 days ago

Verified 4721a6e5

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

a-googler committed 100 days ago

21d1289e

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

a-googler committed 100 days ago

8b8b9298

Fixed broken nnx.statelib.diff

vfdev-5 committed 101 days ago

11a42640

Newer Older