google/flax

Pull Requests Commits

add numpy scalar serialization

jheek committed 5 years ago

f8cd315b

Merge pull request #583 from google:BertrandRdp-readme-linen

a-googler committed 5 years ago

84674691

Merge pull request #582 from google/rtd-linen

levskaya committed 5 years ago

Verified 587bc1f9

improved Embed integer check

nestordemeure committed 5 years ago

Verified d523e883

Merge pull request #585 from levskaya:serialize_bf16_fix

a-googler committed 5 years ago

a28f7c23

Fix jax bfloat16 array handling in msgpack serialization.

levskaya committed 5 years ago

0bdbc173

Update README to linen

BertrandRdp committed 5 years ago

Verified 9ede72ae

Clarify that linen docstrings aren't ready yet

avital committed 5 years ago

d7d56256

Link to Linen reference docs from RTD; specify that `flax.nn` is deprecated

avital committed 5 years ago

94feaf5e

Merge pull request #581 from google:avital-linen-out-of-alpha

a-googler committed 5 years ago

8d7c3947

Graduate Linen out of alpha

avital committed 5 years ago

Verified 9865a222

Merge pull request #560 from jheek:input-checks

a-googler committed 5 years ago

18d82dd4

remove redundant assert

jheek committed 5 years ago

a83048dd

improved Embed input type check

nestordemeure committed 5 years ago

Verified b8ad7097

add the silu activation function

TFUsers committed 5 years ago

a6f4e814

Merge pull request #508 from wrzadkow:ppo-linen

a-googler committed 5 years ago

d860ba16

Removing unused import.

freddyaboulton committed 5 years ago

837bffab

Add support for single input Conv and ConvTranspose in linen.

freddyaboulton committed 5 years ago

02d49dc2

Allow restoring raw state_dict using checkpoints.restore_checkpoint().

jpuigcerver committed 5 years ago

c1d00aea

Fix linen wmt example cache init handling bug.

levskaya committed 5 years ago

8ed5af59

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

Updated notebook run from scratch

Bertrand Rondepierre committed 5 years ago

f200269b

Updating flax part following up on review

Bertrand Rondepierre committed 5 years ago

ef4b1eff

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

Newer Older