google/flax

Pull Requests Commits

remove freeze in apply

jheek committed 5 years ago

4924ce77

add scope path to error messages

jheek committed 5 years ago

26f14b5a

Merge pull request #557 from jheek:harden-frozen-dict-impl

a-googler committed 5 years ago

b5a6aebe

Input checks for apply

jheek committed 5 years ago

0609255a

Make FrozenDict constructor freeze correctly.- Makes freeze a synonym of the FrozenDict constructor- Optimize freezing FrozenDicts by sharing immutable internal state.

jheek committed 5 years ago

6d0133f5

Merge pull request #538 from jheek:Remove-reservation-hack

a-googler committed 5 years ago

154a34fe

Merge pull request #556 from jheek:fix-frozen-dict-restore

a-googler committed 5 years ago

929395cf

Merge pull request #555 from levskaya:trafofix

a-googler committed 5 years ago

afec882d

Avoid nested FrozenDic during state restore.

jheek committed 5 years ago

e37b0a34

Fix transient module-state handling by transforms.

levskaya committed 5 years ago

009ce960

Merge pull request #553 from jheek:test-flax-profile

a-googler committed 5 years ago

fdee7e67

Test with named_call active

jheek committed 5 years ago

b1069ce6

Use daemon processes & disable TF GPU

wrzadkow committed 5 years ago

fbf90763

use core reservation check in linen.

jheek committed 5 years ago

94b8017a

Reduce the usage of static-argnums, in particular with non-hashable objects.

jblespiau committed 5 years ago

bb5927c8

Merge pull request #526 from bkkaggle:master

a-googler committed 5 years ago

bcda8ddd

Fix ModelParamTraversal removes empty dicts

jheek committed 5 years ago

23f6b95a

Merge pull request #549 from jheek:fix-empty-node-traversion

a-googler committed 5 years ago

a3fd2e2b

Preserve nested dict in MultiOptimizer

jheek committed 5 years ago

2a1f86a0

Merge pull request #493 from andsteing:mnist

a-googler committed 5 years ago

2bbdb288

Merge pull request #529 from jheek:improved-mutable-collections

a-googler committed 5 years ago

fa536311

Refactor collection creation to fail early when modifying an undefined collection.Before an non-existing non-mutable collection would just be silently ignored.

jheek committed 5 years ago

45fafe6a

Merge pull request #530 from jheek:param-shape-check

a-googler committed 5 years ago

d9c4faf5

Merge pull request #534 from levskaya:modfix

a-googler committed 5 years ago

6c0088d8

Merge pull request #541 from jheek:ppo-daemon-fix

a-googler committed 5 years ago

5c08bba1

let agens run in daemon proccess

jheek committed 5 years ago

859a5317

jheek committed 5 years ago

2807ad40

Check shape consistency for paramsAdd shape check using `jax.eval_shape` to validate that the return value of init_fn has the same shape as the exisitng paramater.

jheek committed 5 years ago

0a8ee656

Merge pull request #536 from marcvanzee:documentation

a-googler committed 5 years ago

ae2e4463

Minor improvements in dtype handling, broadcast option for dropout.

levskaya committed 5 years ago

c2865a17

Newer Older