search

google / trax

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

[JAX] Remove unused test utilities that access JAX internals that are about to be removed. #1736

Closed copybara-service[bot] closed 2 years ago

copybara-service[bot] commented 2 years ago

[JAX] Remove unused test utilities that access JAX internals that are about to be removed.