xla
Showcase jax.grad in torch_xla
#8800
Open

Showcase jax.grad in torch_xla #8800

zpcore wants to merge 3 commits into master from piz/imp_test
zpcore
zpcore show case use jax.grad in torch_xla
8f231bdd
zpcore zpcore requested a review from qihqi qihqi 1 year ago
zpcore zpcore requested a review from tengyifei tengyifei 1 year ago
zpcore zpcore changed the title show case use jax.grad in torch_xla Showcase jax.grad in torch_xla 1 year ago
zpcore nit
f2453642
tengyifei
tengyifei commented on 2025-03-06
zpcore split the grad test out
bf42d3c4
zpcore zpcore marked this pull request as ready for review 1 year ago
tengyifei
tengyifei requested changes on 2025-03-06

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone