google/flax

Pull Requests Commits

cgarciae committed 2 years ago

c1cbea74

Merge pull request #3367 from chiamp:error

a-googler committed 2 years ago

242f84ca

Merge pull request #3368 from 8bitmp3:add-maxtext

a-googler committed 2 years ago

406116c1

Add MaxText open source LLM to index.rst

8bitmp3 committed 2 years ago

Verified 5d4f3b7d

chiamp committed 2 years ago

2e458817

Merge pull request #3365 from chiamp:bind

a-googler committed 2 years ago

7dba76cd

Merge pull request #3335 from chiamp:spectral_norm

a-googler committed 2 years ago

b05cbfa4

fixed bind-unbind bug

chiamp committed 2 years ago

0d9862e5

Merge pull request #3357 from google:dont-propagate-default-args-in-tabulate

a-googler committed 2 years ago

4d8b4aff

Merge pull request #3359 from chiamp:filter_warning

a-googler committed 2 years ago

fd99fdb2

make DenyList('intermediates') the default for mutable

cgarciae committed 2 years ago

fbc0207b

don't propagate default arguments from .tabulate to .init

cgarciae committed 2 years ago

9c9218d7

Merge pull request #3356 from google:disable-pyink

a-googler committed 2 years ago

63420d35

cgarciae committed 2 years ago

b01402e3

cgarciae committed 2 years ago

04e64f9b

added spectral norm

chiamp committed 2 years ago

15f47bf6

removed DeprecationWarning filter

chiamp committed 2 years ago

4d5ab7ab

Merge pull request #3354 from chiamp:dunder

a-googler committed 2 years ago

d33a33e6

Fix custom prng dtype check

Jake VanderPlas committed 2 years ago

79915d26

use jax.Array type for rng keys

chiamp committed 2 years ago

dbc92549

Merge pull request #3358 from IvyZX:push

a-googler committed 2 years ago

ae402f3c

Make ckpt path absolute

IvyZX committed 2 years ago

1b52b9d3

Merge pull request #3352 from chiamp:chex_warning

a-googler committed 2 years ago

35fcd45c

ignore chex import warning

chiamp committed 2 years ago

070c8be7

Merge pull request #3306 from chiamp:cursor_find

a-googler committed 2 years ago

7f279320

Turn OCDBT on by default

ChromeHearts committed 2 years ago

a2a1eff3

Return sharding whenever the leaf has .sharding attribute, not checking explicitly for jax.Array.

IvyZX committed 2 years ago

2f2ec1a0

Merge pull request #3342 from chiamp:flax_basics

a-googler committed 2 years ago

caba06f8

Merge pull request #3344 from chiamp:resnet

a-googler committed 2 years ago

ed2b1cf6

Update lift to use the new PRNGKey check to suppress deprecation warnings

a-googler committed 2 years ago

830d3350

Older