Trax — Deep Learning with Clear Code and Speed
https://github.com/google/trax
531 forks.
5,599 stars.
71 open issues.
Recent commits:
- Adding support for rationale order prediction in MathQA tasks.PiperOrigin-RevId: 353545244, Copybara-Service
- A gin file with mixtures of glue tasks defined using the Parallel method.PiperOrigin-RevId: 353545145, Copybara-Service
- Cache K/V in predict mode in Encoder-Decoder attention.PiperOrigin-RevId: 353530559, Copybara-Service
- Fix predict mode in LocallyConvDense and add test for predict mode.PiperOrigin-RevId: 353520229, Copybara-Service
- Add save_to_file method for Layer.PiperOrigin-RevId: 353519126, Copybara-Service
Was this post helpful?
Let us know if you liked the post. That’s the only way we can improve.