search

google / trax

Trax — Deep Learning with Clear Code and Speed
Apache License 2.0
8.07k stars 813 forks source link

Remove dependence on private JAX test utilities #1737

Closed copybara-service[bot] closed 2 years ago

copybara-service[bot] commented 2 years ago

Remove dependence on private JAX test utilities