google/flax

Pull Requests Commits

Fix or ignore some pytype errors.

rchen152 committed 5 years ago

ce30a8ec

Merge pull request #706 from avital:param-names

a-googler committed 5 years ago

9a0b06d2

Relax parameter attribute check

avital committed 5 years ago

ec995477

Merge pull request #680 from jheek:lift-md

a-googler committed 5 years ago

ffaf51e7

Move notes into docs

jheek committed 5 years ago

b7cd0432

Merge pull request #647 from qiuminxu:master

a-googler committed 5 years ago

06a79634

Merge pull request #694 from andsteing:launch_gce

a-googler committed 5 years ago

96c3cbd7

Update setup.py

qiuminxu committed 5 years ago

e0365c08

Install jaxlib+cuda if ACCELERATOR_TYPE =~ nvidia-

andsteing committed 5 years ago

36ca83ac

Adds clu to imagenet/requirements.txt

andsteing committed 5 years ago

e8e2a986

Update README.md

avital committed 5 years ago

Verified 7ef53e6e

Merge pull request #681 from 8bitmp3:patch-1

a-googler committed 5 years ago

b275ff5e

Merge pull request #657 from avital:docs2

a-googler committed 5 years ago

567dcfcb

Replace flax named_call with upstream version jax.named_call

qiuminxu committed 5 years ago

bcf4eefb

Update flax/nn/base.py

8bitmp3 committed 5 years ago

Verified e875ef8e

Respond to @andstein's comments

avital committed 5 years ago

5e0ec167

Always start execution from main in benchmarks.

Mohit Reddy committed 5 years ago

498d7275

Merge remote-tracking branch 'upstream/master' into launch_gce

andsteing committed 5 years ago

b5df8d29

Rename -gpu_type to -accelerator_type, better help

andsteing committed 5 years ago

596c0b93

Updates README.md to respond to avital@'s comments

andsteing committed 5 years ago

3978a694

Merge pull request #702 from jheek/fix-flax-docs-install

jheek committed 5 years ago

Verified 433e748a

add correct doc install command

jheek committed 5 years ago

f13dbb13

Merge pull request #696 from jheek:fix-transform-setup

a-googler committed 5 years ago

c36dd738

Re-add flax install to doc requirementtts

jheek committed 5 years ago

2e9c4be3

Avoid a NumPy bug, triggered by the JAX change

mattjj committed 5 years ago

69b9d44f

Remove default value for --config flag.

Marvin182 committed 5 years ago

0bed2f65

Updating readme with quick install instructions

a-googler committed 5 years ago

99c8f1e7

Benchmark method was previously changed, but missed few occurrences.

Mohit Reddy committed 5 years ago

1aff5509

Merge pull request #698 from henry-prior/master

avital committed 5 years ago

Verified 5670ffb2

Move JAX-RL community example from deprecated section to Linen section

henry-prior committed 5 years ago

70571d21

Older