Skip to content

Request: to open-source the torch_directml native extension now that DirectML is in maintenance mode #729

@yanite

Description

@yanite

Hi Microsoft team,
Thank you for your work on DirectML and for making it available to the community. As noted in the repository, "DirectML is in maintenance mode", which implies that the project is no longer under active feature development but will continue to be supported.
Given this status, I’d like to respectfully request that the native Python extension source code (i.e., the C++/pybind11 layer that builds torch_directml_native.*.pyd) be made publicly available.
Currently, only the precompiled .pyd binaries are distributed via PyPI, while the binding code that connects PyTorch tensors to DirectML operators remains closed. This creates several practical limitations:

  • Users cannot rebuild the package for custom environments (e.g., specific Python versions, embedded systems, or air-gapped deployments).
  • Debugging, auditing, or extending the backend (e.g., adding experimental operators or modifying tensor layout handling) is impossible.
  • The project cannot be considered truly open source if a core component—the Python-native bridge—is not available for inspection or contribution.

Since Microsoft has already open-sourced the DMLX operator library, headers, and samples, releasing the pybind11 binding layer would align with the spirit of open collaboration—especially for a project now in maintenance mode, where community contributions could help sustain usability and portability.
Open-sourcing this final piece would greatly empower researchers, enterprise users, and developers who rely on DirectML as a cross-vendor GPU inference solution on Windows.
Thank you for considering this request!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions