Skip to content

[WRONG PR] @assume_pure#8923

Merged
qihqi merged 7 commits intoyifeit/call-jax-out-pytreefrom
yifeit/vjp-in-xla
Apr 10, 2025
Merged

[WRONG PR] @assume_pure#8923
qihqi merged 7 commits intoyifeit/call-jax-out-pytreefrom
yifeit/vjp-in-xla

Conversation

@tengyifei
Copy link
Copy Markdown
Collaborator

@tengyifei tengyifei commented Apr 2, 2025

This is superseded by: #8962

@tengyifei tengyifei changed the title Yifeit/vjp in xla @assume_pure Apr 2, 2025
@tengyifei tengyifei force-pushed the yifeit/torchax-in-torch-xla branch from 13a631b to 8621943 Compare April 4, 2025 05:55
@tengyifei tengyifei force-pushed the yifeit/vjp-in-xla branch from 4e63515 to a1230e1 Compare April 9, 2025 00:38
@tengyifei tengyifei changed the base branch from yifeit/torchax-in-torch-xla to master April 9, 2025 00:38
@tengyifei tengyifei force-pushed the yifeit/vjp-in-xla branch from c37e821 to 422c6df Compare April 9, 2025 07:52
@tengyifei tengyifei marked this pull request as ready for review April 9, 2025 07:52
@tengyifei tengyifei changed the base branch from master to yifeit/call-jax-out-pytree April 10, 2025 06:31
@qihqi qihqi merged commit 4399a08 into yifeit/call-jax-out-pytree Apr 10, 2025
23 of 24 checks passed
@tengyifei tengyifei changed the title @assume_pure [WRONG PR] @assume_pure Apr 11, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Use jax autograd from PyTorch

2 participants