google/flax

Pull Requests Commits

[flax] Remove redundant `deterministic` parameter from `dot_product_attention`.

chr1sj0nes committed 2 years ago

da82965a

Merge pull request #3356 from google:disable-pyink

a-googler committed 2 years ago

63420d35

cgarciae committed 2 years ago

b01402e3

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

updated flax_basics

chiamp committed 2 years ago

7367002e

changed resnet v1 to v1.5

chiamp committed 2 years ago

69b37323

Merge pull request #3343 from chiamp:haiku_upgrade

a-googler committed 2 years ago

5d846a59

updated haiku guide with new JAX RNG api

chiamp committed 2 years ago

089f237b

Migrate flax from using old-style PRNG keys to new-style typed PRNG keys

Jake VanderPlas committed 2 years ago

654ae1a4

Merge pull request #3277 from IvyZX:pen

a-googler committed 2 years ago

d9f83bf7

Merge pull request #3334 from google:0.7.5

a-googler committed 2 years ago

302a7fa9

Fix issue with rng check for new-style PRNG keys

Jake VanderPlas committed 2 years ago

d6f744a4

cgarciae committed 2 years ago

33b460cc

Merge pull request #3331 from levskaya:version_fix

a-googler committed 2 years ago

9c40e030

Update minimum python version info.

levskaya committed 2 years ago

7ce981bc

Bump version no by 1 post-release.

IvyZX committed 2 years ago

ca3ea06f

Merge pull request #3326 from IvyZX:push

a-googler committed 2 years ago

5f06f08c

IvyZX committed 2 years ago

ad8d22e9

Older