google/flax

Pull Requests Commits

Refactor ArrayCheckpointHandler to use BasePyTreeCheckpointHandler.

a-googler committed 356 days ago

4855c5cd

Merge pull request #4857 from nenuadrian:patch-1

a-googler committed 356 days ago

a1451a0e

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

a-googler committed 357 days ago

5d776fb2

document jax.random shorthand methods

cgarciae committed 358 days ago

10398772

Merge pull request #4893 from IvyZX:mesh-ctx

a-googler committed 358 days ago

fffc4db8

Merge pull request #4894 from google:find-duplicates

a-googler committed 359 days ago

ee634f5d

add find_duplicates

cgarciae committed 360 days ago

916c0325

Sharding API improvements (non breaking)

IvyZX committed 363 days ago

3aa32309

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

marksandler2 committed 363 days ago

65d54a95

Merge pull request #4892 from IvyZX:sp-fix

a-googler committed 363 days ago

00506153

Pin sentencepiece version to 0.2.0 to fix head

IvyZX committed 363 days ago

346ecb5e

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

a-googler committed 363 days ago

5dcc888f

Merge pull request #4880 from codinfox:update-typo

a-googler committed 363 days ago

4a13edd3

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

Cristian Garcia committed 364 days ago

ded9094c

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

AlexeyG committed 1 year ago

7a429f33

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

a-googler committed 1 year ago

af43b6a9

Internal change

IvyZX committed 1 year ago

62e07341

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

a-googler committed 1 year ago

58ddd259

Merge pull request #4884 from IvyZX:docs-dataclass

a-googler committed 1 year ago

2f32fc7e

Merge pull request #4885 from IvyZX:docs-readme

a-googler committed 1 year ago

e48c07ea

Update doc build instructions and clean up unused packages

IvyZX committed 1 year ago

a8148c4b

Improve docs related with dataclasses

IvyZX committed 1 year ago

f99b0267

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

a-googler committed 1 year ago

087c95a7

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

a-googler committed 1 year ago

0a5f67cd

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

jlperla committed 1 year ago

Verified ae86d5e6

Fix whitespace doctest error

IvyZX committed 1 year ago

a58b2d9d

Update typo in nnx.Optimizer

codinfox committed 1 year ago

Verified 8393c6da

Merge pull request #4876 from google:rngs-helpers

a-googler committed 1 year ago

38cd9083

Merge pull request #4878 from google:ver-0.11.1

a-googler committed 1 year ago

a0a291cf

updates to version 0.11.1

cgarciae committed 1 year ago

2a4a93a5

Older