Trax — Deep Learning with Clear Code and Speed
https://github.com/google/trax
434 forks.
4,975 stars.
31 open issues.
Recent commits:
- Block sparse feedforward layerPiperOrigin-RevId: 334887634, Copybara-Service
- Add a very quick-to-execute test for Reformer2PiperOrigin-RevId: 334843929, Copybara-Service
- Fix reversible layer after google/jax#4039:differentiating wrt booleans will return float0s which are not compatible with arithmetic operations.PiperOrigin-RevId: 334728148, Copybara-Service
- [TF-numpy] Adds XLA test to lax_numpy_test.PiperOrigin-RevId: 334461210, Copybara-Service
- [TF-numpy/extensions] Moves the single-output-as-list workaround for tpu.replicate to within tf.function, because accessing the global variable _orig_result_is_list outside tf.function (as we do now) is not thread-safe.PiperOrigin-RevId: 334446985, Copybara-Service
Was this post helpful?
Let us know if you liked the post. That’s the only way we can improve.