Download Latest Version
v1.4.1.zip (116.8 MB)
Get an email when there's a new version of Trax
| Name | Modified | Size | Downloads / Week |
|---|---|---|---|
| Parent folder | |||
| README.md | 2020-12-18 | 1.7 kB | |
| v1.3.7.tar.gz | 2020-12-18 | 84.3 MB | |
| v1.3.7.zip | 2020-12-18 | 84.6 MB | |
| Totals: 3 Items | 168.9 MB | 0 | |
Lots of documentation, bugs squashed and misc changes.
Features - Load checkpoint for fine-tuning in https://github.com/google/trax/commit/0349c3f2c32ed83d0ba54a0f6652766ea5467cca by @henrykmichalewski ! - Added generic input pipeline for GLUE tasks in https://github.com/google/trax/commit/2f490e83a9be1b802dccf866c52c8908645e10d6 by @henrykmichalewski - thanks a lot! - Some nascent support for bfloat16s!
Models - Performer's Favor and CausalFavor - https://github.com/google/trax/commit/77db199392ff967e49156ccb24c916b270b47eca thanks @lukaszkaiser ! - Funnel-Transformer in https://github.com/google/trax/pull/1156 thanks a lot @mvxxx ! - BERT for Trax in https://github.com/google/trax/pull/1254 , https://github.com/google/trax/pull/1223, etc by @piotrekp1 - thanks a lot! - Residual Exchange Network by @kkanska in https://github.com/google/trax/commit/3a8f402b722a0d9ae934ed399e6532dba8401f9e ! Thanks!
PRs Merged
- Fixing broken example links in https://github.com/google/trax/pull/1263 thanks @amtagrwl !
- Added WideResnet, Deconv etc Example Notebook in https://github.com/google/trax/pull/1259 , https://github.com/google/trax/pull/1202, https://github.com/google/trax/pull/1232 thanks a lot! @SauravMaheshkar !
- Remove implicit object from the base class in https://github.com/google/trax/pull/1228 thanks @HarshCasper !
- Fashion MNIST example in https://github.com/google/trax/pull/1199 thanks @Jimexist !
- Fix PretrainedBERT init in https://github.com/google/trax/pull/1135 by @hepaajan !
- Typo in the TransformerDecoder input parameters description in https://github.com/google/trax/pull/1100 by @kujaomega !