Trax — Deep Learning with Clear Code and Speed
https://github.com/google/trax
647 forks.
6,489 stars.
82 open issues.
Recent commits:
- Merge pull request #1696 from lukaszkaiser:lkPiperOrigin-RevId: 396062436, Copybara-Service
- Adding handling of GPU clusters and recent JAX renames., Lukasz Kaiser
- Make T5 imports dynamic so it can be optional., Lukasz Kaiser
- Corrects link to TensorFlow NumPy in README.md, GitHub
- Ensure values passed to jax.numpy functions are arrays rather than lists.Why? This will soon be a requirement in JAX; see https://github.com/google/jax/issues/7737PiperOrigin-RevId: 394106413, Copybara-Service
Was this post helpful?
Let us know if you liked the post. That’s the only way we can improve.