trax: avoid referencing deprecated jax.random.threefry_2x32 - Githubissues

google / trax

Trax — Deep Learning with Clear Code and Speed

Apache License 2.0

8.01k stars 813 forks source link

trax: avoid referencing deprecated jax.random.threefry_2x32 #1789

Closed copybara-service[bot] closed 8 months ago

copybara-service[bot] commented 8 months ago

trax: avoid referencing deprecated jax.random.threefry_2x32