Skip to content

kernels: add support for Neuron and NKI#285

Draft
danieldk wants to merge 6 commits intomainfrom
neuron-nki
Draft

kernels: add support for Neuron and NKI#285
danieldk wants to merge 6 commits intomainfrom
neuron-nki

Conversation

@danieldk
Copy link
Member

This change add support for Torch Neuron and NKI to the kernels Python package. Support for builder is not added in this change.

@HuggingFaceDocBuilderDev

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

@danieldk danieldk marked this pull request as ready for review February 16, 2026 17:03
MekkCyber
MekkCyber previously approved these changes Feb 16, 2026
Copy link
Collaborator

@MekkCyber MekkCyber left a comment

Choose a reason for hiding this comment

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

A new backend! let's go 🔥

Comment on lines +97 to +98
Build vaariant for arch kernels, returns `None` when the backend
does not (yet) support arch kernels.
Copy link
Collaborator

Choose a reason for hiding this comment

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

Suggested change
Build vaariant for arch kernels, returns `None` when the backend
does not (yet) support arch kernels.
Build variant for arch kernels, returns `None` when the backend
does not (yet) support arch kernels.

@danieldk danieldk marked this pull request as draft February 17, 2026 14:36
@danieldk
Copy link
Member Author

Putting this in draft for a bit, the Neuron detection code needs te be changed.

Copy link

@JingyaHuang JingyaHuang left a comment

Choose a reason for hiding this comment

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

thx for taking care of this! look good to me, except for the coming transition from torch xla. also wondering if you plan to add a ci with trn2 runner.

@danieldk
Copy link
Member Author

thx for taking care of this! look good to me, except for the coming transition from torch xla. also wondering if you plan to add a ci with trn2 runner.

Hooking up CI would be great!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants

Comments