mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-22 21:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
25,133 Commits 508 Branches 263 Tags
Commit Graph

2 Commits

Author SHA1 Message Date
Ivy Zheng
26c40fadfd Add jax.tree shortcuts for .*_with_path calls, for convenience of users. 
PiperOrigin-RevId: 705645570
 2024-12-12 15:13:32 -08:00
Jake VanderPlas
6934a4b76b Add jax.tree module with aliases of jax.tree_util 2024-02-12 13:07:59 -08:00