Skip to content

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

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open

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

wants to merge 1 commit into from

Conversation

Steboss
Copy link
Contributor

@Steboss Steboss commented May 22, 2025

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

@Steboss Steboss requested review from ruomingp, markblee and a team as code owners May 22, 2025 09:33
@Steboss Steboss mentioned this pull request May 23, 2025
Closed
@changlan
Copy link
Member

Hi @Steboss are you still working on this PR?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@markblee markblee Awaiting requested review from markblee

1 more reviewer

@ruomingp ruomingp ruomingp approved these changes

Reviewers whose approvals may not affect merge requirements

At least 1 approving review is required to merge this pull request.

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
3 participants
@Steboss @changlan @ruomingp