xla
[torch_xla2] Add oss only tags to jax env initialization
#7920
Merged

[torch_xla2] Add oss only tags to jax env initialization #7920

qihqi merged 1 commit into master from chunnienc/fix-g3-usage
chunnienc
chunnienc Update __init__.py
65b0aa21
chunnienc chunnienc requested a review from qihqi qihqi 1 year ago
qihqi
qihqi approved these changes on 2024-08-28
qihqi qihqi merged eedf2b9b into master 1 year ago
ManfeiBai ManfeiBai added torchxla2

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone