Skip to content
Navigation Menu
Sign in
Appearance settings
Platform
AI CODE CREATION
GitHub Copilot
Write better code with AI
GitHub Copilot app
Direct agents from issue to merge
MCP Registry
Integrate external tools
DEVELOPER WORKFLOWS
Actions
Automate any workflow
Codespaces
Instant dev environments
Issues
Plan and track work
Code Review
Manage code changes
Code Quality
Enforce quality at merge
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
View all resources
Open Source
COMMUNITY
GitHub Sponsors
Fund open source developers
PROGRAMS
Security Lab
Maintainer Community
Accelerator
GitHub Stars
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
Type
/
to search
Sign in
Sign up
Appearance settings
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 }}
Uh oh!
There was an error while loading.
Please reload this page
.
NVIDIA
/
TransformerEngine
Public
Notifications
You must be signed in to change notification settings
Fork
799
Star
3.5k
Code
Issues
159
Pull requests
162
Discussions
Actions
Security and quality
0
Insights
Additional navigation options
Code
Issues
Pull requests
Discussions
Actions
Security and quality
Insights
[PyTorch] Support pickling Float8Tensor
- #529
#529
Merged
timmoon10
merged 13 commits into
NVIDIA:main
NVIDIA/TransformerEngine:main
from
timmoon10:float8tensor-pickle
timmoon10/TransformerEngine:float8tensor-pickle
Copy head branch name to clipboard
Dec 7, 2023
Conversation
Commits
13
(13)
Checks
Files changed
Merged
[PyTorch] Support pickling Float8Tensor
#529
timmoon10
merged 13 commits into
NVIDIA:main
NVIDIA/TransformerEngine:main
from
timmoon10:float8tensor-pickle
timmoon10/TransformerEngine:float8tensor-pickle
Copy head branch name to clipboard
Commits
Commits on Nov 19, 2023
Float8Tensor uses cached transpose if available
Show description for 5215774
timmoon10
committed
5215774
View commit details
Copy full SHA for 5215774
Browse repository at this point
Commits on Nov 20, 2023
Fix bug with non-2D transpose
Show description for 22eccf6
timmoon10
committed
22eccf6
View commit details
Copy full SHA for 22eccf6
Browse repository at this point
Merge branch 'main' into float8tensor-transpose-caching
timmoon10
authored
0dcd8f4
View commit details
Copy full SHA for 0dcd8f4
Browse repository at this point
Commits on Nov 21, 2023
Custom pickling for Float8Tensor
Show description for 30f2805
timmoon10
committed
30f2805
View commit details
Copy full SHA for 30f2805
Browse repository at this point
Debug test for pickling Float8Tensor
Show description for d2470cc
timmoon10
committed
d2470cc
View commit details
Copy full SHA for d2470cc
Browse repository at this point
Merge branch 'main' into float8tensor-transpose-caching
timmoon10
authored
57c7e03
View commit details
Copy full SHA for 57c7e03
Browse repository at this point
Merge branch 'float8tensor-transpose-caching' into float8tensor-pickle
timmoon10
committed
76ea1d1
View commit details
Copy full SHA for 76ea1d1
Browse repository at this point
Fix merge conflict
Show description for 871eaf0
timmoon10
committed
871eaf0
View commit details
Copy full SHA for 871eaf0
Browse repository at this point
Commits on Dec 1, 2023
Review suggestions from @sudhakarsingh27
Show description for 68d0ca9
timmoon10
committed
68d0ca9
View commit details
Copy full SHA for 68d0ca9
Browse repository at this point
Merge branch 'main' into float8tensor-pickle
timmoon10
authored
60ec28f
View commit details
Copy full SHA for 60ec28f
Browse repository at this point
Commits on Dec 2, 2023
Merge branch 'main' into float8tensor-pickle
timmoon10
authored
068b64e
View commit details
Copy full SHA for 068b64e
Browse repository at this point
Commits on Dec 5, 2023
Add unit test for checkpointing model with FP8 params
Show description for cd60e0f
timmoon10
committed
cd60e0f
View commit details
Copy full SHA for cd60e0f
Browse repository at this point
Merge branch 'main' into float8tensor-pickle
timmoon10
authored
110b651
View commit details
Copy full SHA for 110b651
Browse repository at this point
You can’t perform that action at this time.