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
798
Star
3.5k
Code
Issues
159
Pull requests
160
Discussions
Actions
Security and quality
0
Insights
Additional navigation options
Code
Issues
Pull requests
Discussions
Actions
Security and quality
Insights
[PyTorch] Expose function to bulk-allocate tensors backed by the same buffer
- #2900
#2900
Merged
timmoon10
merged 14 commits into
NVIDIA:main
NVIDIA/TransformerEngine:main
from
timmoon10:tmoon/group-mlp-bulk-allocate
timmoon10/TransformerEngine:tmoon/group-mlp-bulk-allocate
Copy head branch name to clipboard
May 12, 2026
Conversation
Commits
14
(14)
Checks
Files changed
Merged
[PyTorch] Expose function to bulk-allocate tensors backed by the same buffer
#2900
timmoon10
merged 14 commits into
NVIDIA:main
NVIDIA/TransformerEngine:main
from
timmoon10:tmoon/group-mlp-bulk-allocate
timmoon10/TransformerEngine:tmoon/group-mlp-bulk-allocate
Copy head branch name to clipboard
Commits
Commits on Apr 18, 2026
[PyTorch] Add bulk_allocate utility and use it in quantized tensor allocators
Show description for bde0883
timmoon10
and
claude
committed
bde0883
View commit details
Copy full SHA for bde0883
Browse repository at this point
Bulk-allocate wgrads in grouped linear impls
Show description for 81597a6
timmoon10
committed
81597a6
View commit details
Copy full SHA for 81597a6
Browse repository at this point
[pre-commit.ci] auto fixes from pre-commit.com hooks
Show description for 84096c4
pre-commit-ci[bot]
committed
84096c4
View commit details
Copy full SHA for 84096c4
Browse repository at this point
Merge branch 'main' into tmoon/group-mlp-bulk-allocate
timmoon10
authored
d1d27ee
View commit details
Copy full SHA for d1d27ee
Browse repository at this point
Commits on Apr 22, 2026
Apply review suggestions
Show description for d4ba30b
timmoon10
committed
d4ba30b
View commit details
Copy full SHA for d4ba30b
Browse repository at this point
Nits from Claude
Show description for f73c4f8
timmoon10
committed
f73c4f8
View commit details
Copy full SHA for f73c4f8
Browse repository at this point
Merge branch 'main' into tmoon/group-mlp-bulk-allocate
timmoon10
authored
51ab3cf
View commit details
Copy full SHA for 51ab3cf
Browse repository at this point
[pre-commit.ci] auto fixes from pre-commit.com hooks
Show description for e25ba48
pre-commit-ci[bot]
committed
e25ba48
View commit details
Copy full SHA for e25ba48
Browse repository at this point
Fix incorrect call to `bulk_allocate`
Show description for 16806b4
timmoon10
committed
16806b4
View commit details
Copy full SHA for 16806b4
Browse repository at this point
Commits on May 11, 2026
Merge branch 'main' into tmoon/group-mlp-bulk-allocate
Show description for 6f14edc
timmoon10
committed
6f14edc
View commit details
Copy full SHA for 6f14edc
Browse repository at this point
Fix ambiguous return type
Show description for b1ae2ba
timmoon10
committed
b1ae2ba
View commit details
Copy full SHA for b1ae2ba
Browse repository at this point
[pre-commit.ci] auto fixes from pre-commit.com hooks
Show description for e98d476
pre-commit-ci[bot]
committed
e98d476
View commit details
Copy full SHA for e98d476
Browse repository at this point
Commits on May 12, 2026
Use c10::Device consistently
Show description for 481a91e
timmoon10
committed
481a91e
View commit details
Copy full SHA for 481a91e
Browse repository at this point
Merge branch 'main' into tmoon/group-mlp-bulk-allocate
timmoon10
authored
0057d2b
View commit details
Copy full SHA for 0057d2b
Browse repository at this point
You can’t perform that action at this time.
![pre-commit-ci[bot]](https://avatars.githubusercontent.com/in/68672?v=4&size=32){kind=avatar}