google/flax

Pull Requests Commits

Update optimizer.py to support masked variable from optax.

ywrt committed 166 days ago

Verified 2f213b54

Merge pull request #4857 from nenuadrian:patch-1

a-googler committed 167 days ago

a1451a0e

Merge pull request #4899 from google:jax-random-methods-docs

a-googler committed 169 days ago

5d776fb2

document jax.random shorthand methods

cgarciae committed 169 days ago

10398772

Merge pull request #4893 from IvyZX:mesh-ctx

a-googler committed 169 days ago

fffc4db8

Merge pull request #4894 from google:find-duplicates

a-googler committed 170 days ago

ee634f5d

add find_duplicates

cgarciae committed 171 days ago

916c0325

Sharding API improvements (non breaking)

IvyZX committed 174 days ago

3aa32309

Relax duplicate check to exclude non-string values such as PartitionSpec.UNCONSTRAINED, since those can be repeated.

marksandler2 committed 174 days ago

65d54a95

Merge pull request #4892 from IvyZX:sp-fix

a-googler committed 174 days ago

00506153

Pin sentencepiece version to 0.2.0 to fix head

IvyZX committed 174 days ago

346ecb5e

Merge pull request #4846 from google:enable-split-rngs-docstring

a-googler committed 174 days ago

5dcc888f

Merge pull request #4880 from codinfox:update-typo

a-googler committed 174 days ago

4a13edd3

call jax's source_info_util.register_exclusion in flax's traceback_util.register_exclusion

Cristian Garcia committed 175 days ago

ded9094c

Remove the need for Conv and ConvTranspose to know the precise batch size.

AlexeyG committed 176 days ago

7a429f33

Merge pull request #4860 from mohsinm-dev:fix-replace-by-pure-dict-string-keys

a-googler committed 176 days ago

af43b6a9

Internal change

IvyZX committed 176 days ago

62e07341

Merge pull request #4855 from mohsinm-dev:fix-contributing-docs-redirect

a-googler committed 176 days ago

58ddd259

Merge pull request #4884 from IvyZX:docs-dataclass

a-googler committed 177 days ago

2f32fc7e

Merge pull request #4885 from IvyZX:docs-readme

a-googler committed 177 days ago

e48c07ea

Update doc build instructions and clean up unused packages

IvyZX committed 177 days ago

a8148c4b

Improve docs related with dataclasses

IvyZX committed 177 days ago

f99b0267

Merge pull request #4852 from mohsinm-dev:fix-tolinen-docstring

a-googler committed 178 days ago

087c95a7

Merge pull request #4850 from jlperla:fix-identity-docs

a-googler committed 178 days ago

0a5f67cd

Merge pull request #1 from IvyZX/fix-doctest-whitespace

jlperla committed 179 days ago

Verified ae86d5e6

Fix whitespace doctest error

IvyZX committed 179 days ago

a58b2d9d

Update typo in nnx.Optimizer

codinfox committed 179 days ago

Verified 8393c6da

Merge pull request #4876 from google:rngs-helpers

a-googler committed 180 days ago

38cd9083

Merge pull request #4878 from google:ver-0.11.1

a-googler committed 180 days ago

a0a291cf

updates to version 0.11.1

cgarciae committed 180 days ago

2a4a93a5

Newer Older