axlearn icon indicating copy to clipboard operation
axlearn copied to clipboard
verified publisher icon apple

[UPDATE JAX API] Modify all the calls to jax.tree

Open Steboss opened this issue 7 months ago • 2 comments

This PR gathers all the related PRs that are working on the jax.tree structure #1202 #1200 #1199

Steboss avatar May 22 '25 09:05 Steboss

Hi @Steboss are you still working on this PR?

changlan avatar Jul 26 '25 00:07 changlan

This pull request has been automatically marked as stale because it has been inactive for 60 days. It will be closed in 7 days if no further activity occurs. If you would like to continue working on this, please remove the stale label or leave a comment.

github-actions[bot] avatar Oct 20 '25 02:10 github-actions[bot]