Trax — Deep Learning with Clear Code and Speed
https://github.com/google/trax
816 forks.
8,098 stars.
122 open issues.
Recent commits:
- disable certain tests in lax_numpy_test when numpy 2.0 is usedPiperOrigin-RevId: 673089320, Copybara-Service
- disable testOp when numpy 2.0 is installedPiperOrigin-RevId: 673019213, Copybara-Service
- fix a few test bitwise op tests for np 2.0PiperOrigin-RevId: 672999286, Copybara-Service
- skip testLinSpace and testLogSpace if numpy 2.0 is usedPiperOrigin-RevId: 666057286, Copybara-Service
- [numpy] Fix test failures under NumPy 2.0.PiperOrigin-RevId: 664492429, Copybara-Service
