google/flax

Pull Requests Commits

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

a-googler committed 139 days ago

5d776fb2

document jax.random shorthand methods

cgarciae committed 140 days ago

10398772

Merge pull request #4893 from IvyZX:mesh-ctx

a-googler committed 140 days ago

fffc4db8

Merge pull request #4894 from google:find-duplicates

a-googler committed 141 days ago

ee634f5d

add find_duplicates

cgarciae committed 142 days ago

916c0325

Sharding API improvements (non breaking)

IvyZX committed 145 days ago

3aa32309

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

marksandler2 committed 145 days ago

65d54a95

Merge pull request #4892 from IvyZX:sp-fix

a-googler committed 145 days ago

00506153

Pin sentencepiece version to 0.2.0 to fix head

IvyZX committed 145 days ago

346ecb5e

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

a-googler committed 145 days ago

5dcc888f

Merge pull request #4880 from codinfox:update-typo

a-googler committed 145 days ago

4a13edd3

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

Cristian Garcia committed 146 days ago

ded9094c

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

AlexeyG committed 146 days ago

7a429f33

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

a-googler committed 147 days ago

af43b6a9

Internal change

IvyZX committed 147 days ago

62e07341

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

a-googler committed 147 days ago

58ddd259

Merge pull request #4884 from IvyZX:docs-dataclass

a-googler committed 148 days ago

2f32fc7e

Merge pull request #4885 from IvyZX:docs-readme

a-googler committed 148 days ago

e48c07ea

Update doc build instructions and clean up unused packages

IvyZX committed 148 days ago

a8148c4b

Improve docs related with dataclasses

IvyZX committed 148 days ago

f99b0267

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

a-googler committed 148 days ago

087c95a7

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

a-googler committed 148 days ago

0a5f67cd

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

jlperla committed 150 days ago

Verified ae86d5e6

Fix whitespace doctest error

IvyZX committed 150 days ago

a58b2d9d

Update typo in nnx.Optimizer

codinfox committed 150 days ago

Verified 8393c6da

Merge pull request #4876 from google:rngs-helpers

a-googler committed 151 days ago

38cd9083

Merge pull request #4878 from google:ver-0.11.1

a-googler committed 151 days ago

a0a291cf

updates to version 0.11.1

cgarciae committed 151 days ago

2a4a93a5

Merge pull request #4868 from google:fix-docs-1

a-googler committed 152 days ago

52ddf006

Merge pull request #4869 from google:update-for-dict

a-googler committed 152 days ago

f9a6055b

Newer Older