mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-25 13:46:08 +00:00
Code Issues Packages Projects Releases Wiki Activity
26,252 Commits 508 Branches 263 Tags
Commit Graph

8 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
3660a7df65 Fix annotation for jax.tree.transpose 2024-07-09 08:06:16 -07:00
Peter Hawkins
7f4ef63cd8 Run pyupgrade --py310-plus. 
Also apply manual fixes to import sorting and unused imports.
 2024-06-26 16:10:18 -04:00
Dan Foreman-Mackey
6d35b109fd Rename "Example" to "Examples" in docstrings. 
This PR updates all docstrings that previously had a section heading
called "Example" and replaces that with "Examples" to be consistent.
 2024-06-21 11:43:16 -04:00
Jake VanderPlas
d82b66f77f Document jax.tree.* directly 2024-06-12 14:01:27 -07:00
Jake VanderPlas
814b32a44b tree_all: add support for is_leaf 2024-06-10 09:46:15 -07:00
Jake VanderPlas
be51345b25 Avoid dynamic docstring generation for jax.tree aliases 2024-06-04 11:56:35 -07:00
Jake VanderPlas
6934a4b76b Add jax.tree module with aliases of jax.tree_util 2024-02-12 13:07:59 -08:00