xla
Include torchax in torch_xla
#8895
Merged
Overview
Commits
13
Changes
View On GitHub

Include torchax in torch_xla #8895

tengyifei merged 13 commits into master from yifeit/torchax-in-torch-xla
tengyifei
tengyifei tengyifei force-pushed the yifeit/scan-shard-as branch from 5f84ac1d to 9dcfe1c3 1 year ago
tengyifei tengyifei changed the base branch from yifeit/scan-shard-as to master 1 year ago
tengyifei tengyifei force pushed from 8eb87caf to a67cc9c7 1 year ago
tengyifei tengyifei changed the title Yifeit/torchax in torch xla torchax in torch xla 1 year ago
tengyifei tengyifei force pushed from 3329c58d to 7c741afe 1 year ago
tengyifei tengyifei force pushed from 8bb02d4e to 4c4ca93f 1 year ago
tengyifei tengyifei changed the title torchax in torch xla Include torchax in torch xla 1 year ago
tengyifei tengyifei marked this pull request as ready for review 1 year ago
tengyifei tengyifei requested a review from mikegre-google mikegre-google 1 year ago
tengyifei tengyifei requested a review from qihqi qihqi 1 year ago
tengyifei tengyifei requested a review from lsy323 lsy323 1 year ago
tengyifei tengyifei requested a review from bhavya01 bhavya01 1 year ago
tengyifei tengyifei requested a review from zpcore zpcore 1 year ago
tengyifei tengyifei changed the title Include torchax in torch xla Include torchax in torch_xla 1 year ago
zpcore
zpcore commented on 2025-04-02
zpcore
zpcore commented on 2025-04-02
qihqi
qihqi approved these changes on 2025-04-04
tengyifei bdist_wheel works but editable install fails
f7008126
tengyifei wip
05fb0cfb
tengyifei Don't reinstall torchax in CI
bb91035f
tengyifei Always run all torch_xla + torchax tests
13b1a160
tengyifei wip
fea69c92
tengyifei Fix Jax nightly install
be470128
tengyifei Find all packages in torchax
f65f06e7
tengyifei Install JAX cuda plugin for triton integration
02f9673d
tengyifei wip
fe78c69e
tengyifei Remove
f61475e9
tengyifei Refactor
08e9e321
tengyifei Add TODO and nullify install_post_deps_pytorch_xla
86219430
tengyifei tengyifei force pushed from 13a631b2 to 86219430 1 year ago
tengyifei split out docs change
64a8cad6
tengyifei tengyifei merged bb5c6a0b into master 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum