Skip to content
Navigation Menu
Toggle navigation
Sign in
Appearance settings
Platform
AI CODE CREATION
GitHub Copilot
Write better code with AI
GitHub Spark
Build and deploy intelligent apps
GitHub Models
Manage and compare prompts
MCP Registry
New
Discover and integrate external tools
DEVELOPER WORKFLOWS
Actions
Automate any workflow
Codespaces
Instant dev environments
Issues
Plan and track work
Code Review
Manage code changes
APPLICATION SECURITY
GitHub Advanced Security
Find and fix vulnerabilities
Code security
Secure your code as you build
Secret protection
Stop leaks before they start
EXPLORE
Why GitHub
Documentation
Blog
Changelog
Marketplace
View all features
Solutions
BY COMPANY SIZE
Enterprises
Small and medium teams
Startups
Nonprofits
BY USE CASE
App Modernization
DevSecOps
DevOps
CI/CD
View all use cases
BY INDUSTRY
Healthcare
Financial services
Manufacturing
Government
View all industries
View all solutions
Resources
EXPLORE BY TOPIC
AI
Software Development
DevOps
Security
View all topics
EXPLORE BY TYPE
Customer stories
Events & webinars
Ebooks & reports
Business insights
GitHub Skills
SUPPORT & SERVICES
Documentation
Customer support
Community forum
Trust center
Partners
Open Source
COMMUNITY
GitHub Sponsors
Fund open source developers
PROGRAMS
Security Lab
Maintainer Community
Accelerator
Archive Program
REPOSITORIES
Topics
Trending
Collections
Enterprise
ENTERPRISE SOLUTIONS
Enterprise platform
AI-powered developer platform
AVAILABLE ADD-ONS
GitHub Advanced Security
Enterprise-grade security features
Copilot for Business
Enterprise-grade AI features
Premium Support
Enterprise-grade 24/7 support
Pricing
Search or jump to...
Search code, repositories, users, issues, pull requests...
Search syntax tips
Provide feedback
Saved searches
Use saved searches to filter your results more quickly
Sign in
Sign up
Appearance settings
Resetting focus
You signed in with another tab or window.
Reload
to refresh your session.
You signed out in another tab or window.
Reload
to refresh your session.
You switched accounts on another tab or window.
Reload
to refresh your session.
Dismiss alert
{{ message }}
pytorch
/
xla
Public
Notifications
You must be signed in to change notification settings
Fork
559
Star
2.7k
Code
Issues
694
Pull requests
189
Discussions
Actions
Projects
11
Security
Uh oh!
There was an error while loading.
Please reload this page
.
Insights
Additional navigation options
Code
Issues
Pull requests
Discussions
Actions
Projects
Security
Insights
Commits
Branch selector
support-optim-shallow-copy
User selector
All users
Datepicker
All time
Commit History
Commits on Jul 31, 2023
Fix lint
janeyx99
committed
2417ef6
Copy full SHA for 2417ef6
Rectify test_zero1.py to account for state_dict modification
janeyx99
committed
f2d7aef
Copy full SHA for f2d7aef
Try fix
janeyx99
committed
1cf7b5c
Copy full SHA for 1cf7b5c
[TEST ONLY] print statements for test_zero1.py to debug
janeyx99
committed
0f139ab
Copy full SHA for 0f139ab
Commits on Jul 28, 2023
[BE] use self.assertEquals instead of str equality in test_zero1.py (#5367)
Show description for 498bce4
janeyx99
authored
498bce4
Copy full SHA for 498bce4
[Dynamo|TPU] Skip `DynamoInferenceBasicTest.test_resnet18` on TPU (#5361)
Show description for 4f4978c
ManfeiBai
authored
4f4978c
Copy full SHA for 4f4978c
Don't rewrite index hints in global save planning (#5348)
jonb377
authored
34b64b0
Copy full SHA for 34b64b0
Add a script for running stablehlo tests. (#5360)
Show description for dbed83c
qihqi
authored
dbed83c
Copy full SHA for dbed83c
Commits on Jul 27, 2023
[Dynamo|TPU] Skip`DynamoTrainingBasicTest.test_resnet18` on TPU (#5362)
Show description for 671bfa7
ManfeiBai
authored
671bfa7
Copy full SHA for 671bfa7
[Dynamo|TPU] Tweak `atol` and `rtol` for `test_dynamo.py` (#5363)
Show description for 8ed67bb
ManfeiBai
authored
8ed67bb
Copy full SHA for 8ed67bb
Revert "[BE] use self.assertEquals instead of str equality in test_zero1.py (#5364)" (#5366)
Show description for 3a81740
JackCaoG
authored
3a81740
Copy full SHA for 3a81740
[BE] use self.assertEquals instead of str equality in test_zero1.py (#5364)
janeyx99
authored
8ada333
Copy full SHA for 8ada333
Check the actual device instead of query env var for virtual device (#5352)
Show description for d09cc69
JackCaoG
authored
d09cc69
Copy full SHA for d09cc69
Add python test for SPMD+Runtime Python API (#5349)
Show description for e70fb29
JackCaoG
authored
e70fb29
Copy full SHA for e70fb29
fix amp dtype setting for GPU (#5337)
Show description for 22ba4af
baoleai
authored
22ba4af
Copy full SHA for 22ba4af
Fix CPU/GPU test coverage (#5355)
Show description for 443fad1
lsy323
authored
443fad1
Copy full SHA for 443fad1
remove unused patches (#5353)
lsy323
authored
1fa34bf
Copy full SHA for 1fa34bf
Add py3.10 whl path to doc, refactor whl table (#5354)
lsy323
authored
b95554b
Copy full SHA for b95554b
Commits on Jul 26, 2023
add fs linker flag (#5347)
lsy323
authored
5673ac4
Copy full SHA for 5673ac4
Make all-reduce a no-op when world size is 1 (#5342)
Show description for 61816b0
will-cromar
authored
61816b0
Copy full SHA for 61816b0
Also dump output sharding on HLO file (#5339)
Show description for efb1e6f
JackCaoG
authored
efb1e6f
Copy full SHA for efb1e6f
Commits on Jul 25, 2023
Check out the release branch instead of origin/master in ansible (#5344)
vanbasten23
authored
68d7fe8
Copy full SHA for 68d7fe8
OpenXLA pin update 07052023 (#5304)
Show description for 8ffbd1a
lsy323
authored
8ffbd1a
Copy full SHA for 8ffbd1a
Update Python device API for SPMD (#5129)
Show description for f996701
JackCaoG
authored
f996701
Copy full SHA for f996701
Commits on Jul 24, 2023
Sharding should be per output of IR Node, instead of per IR Node (#5330)
Show description for 901d154
JackCaoG
authored
901d154
Copy full SHA for 901d154
Commits on Jul 21, 2023
handle single tensor for method send_to_device_single (#5317)
Show description for f5edcb2
JackCaoG
authored
f5edcb2
Copy full SHA for f5edcb2
Commits on Jul 19, 2023
Use `_sharded_cpu_state_dict` functionality to Write Items for SPMD Save Planner (#5315)
Show description for 74d115e
shahyash10
authored
74d115e
Copy full SHA for 74d115e
Enable Some input output aliasing under SPMD (#5320)
JackCaoG
authored
ff8f7a4
Copy full SHA for ff8f7a4
Disable coverage for now (#5321)
JackCaoG
authored
e45cf40
Copy full SHA for e45cf40
Add tf independent serialization (#5308)
Show description for 0b5b519
qihqi
authored
0b5b519
Copy full SHA for 0b5b519
Commits on Jul 18, 2023
quick refactor on _get_group_assignment (#5318)
JackCaoG
authored
a8f723b
Copy full SHA for a8f723b
Fix mismatched GPU docker image in the doc. (#5319)
vanbasten23
authored
42ae0bf
Copy full SHA for 42ae0bf
Support unordered sharding spec for partial replication (#5316)
Show description for 78c6840
JackCaoG
authored
78c6840
Copy full SHA for 78c6840
Update gcc version in CI (#5297)
Show description for e7e1899
lsy323
authored
e7e1899
Copy full SHA for e7e1899
Add docs for AMP (#5313)
Show description for 1cb5a52
cowanmeg
authored
1cb5a52
Copy full SHA for 1cb5a52
Pagination
Previous
Next
You can’t perform that action at this time.
