Skip to content

Add Accelerator API to neural tangent kernels #3578

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Merged
merged 1 commit into from
Sep 19, 2025
Merged

Add Accelerator API to neural tangent kernels #3578

merged 1 commit into from
Sep 19, 2025

Conversation

jafraustro
Copy link
Contributor

@jafraustro jafraustro commented Sep 18, 2025 •
edited
Loading

Motivation

  • Enhance distributed tutorial: integrate Accelerator API for backend selection

Description

  • Integrate the Accelerator API to support multiple accelerators and improve backend initialization.

Checklist

  • The issue that is being fixed is referred in the description (see above "Fixes #ISSUE_NUMBER")
  • Only one issue is addressed in this pull request
  • Labels from the issue that this PR is fixing are added to this pull request
  • No unnecessary issues are included into this pull request.

@jafraustro
Add Accelerator API
e622560 
Signed-off-by: jafraustro <jaime.fraustro.valdez@intel.com>
@pytorch-bot PyTorch Bot
Copy link

pytorch-bot bot commented Sep 18, 2025 •
edited
Loading

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/tutorials/3578

Note: Links to docs will display an error until the docs builds have been completed.

✅ No Failures

As of commit e622560 with merge base fc8981b (image):
💚 Looks good so far! There are no failures yet. 💚

This comment was automatically generated by Dr. CI and updates every 15 minutes.

@meta-cla meta-cla bot added the cla signed label Sep 18, 2025
svekars
svekars approved these changes Sep 19, 2025
View reviewed changes
Copy link
Contributor

@svekars svekars left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@svekars svekars merged commit fd52121 into pytorch:main Sep 19, 2025
21 checks passed
jafraustro added a commit to jafraustro/tutorials that referenced this pull request Oct 1, 2025
@jafraustro
Add Accelerator API to neural tangent kernels (pytorch#3578)
9d19dcb 
- Integrate the Accelerator API to support multiple accelerators and
improve backend initialization.


Signed-off-by: jafraustro <jaime.fraustro.valdez@intel.com>
jafraustro added a commit to jafraustro/tutorials that referenced this pull request Oct 1, 2025
@jafraustro
Add Accelerator API to neural tangent kernels (pytorch#3578)
c6017cd 
- Integrate the Accelerator API to support multiple accelerators and
improve backend initialization.


Signed-off-by: jafraustro <jaime.fraustro.valdez@intel.com>
@jafraustro jafraustro deleted the jafraustro/neural_tangent branch October 1, 2025 18:18
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@svekars svekars svekars approved these changes

Assignees

No one assigned

Labels

cla signed

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

2 participants

@jafraustro @svekars