Pytorch DirectML integration in Pytorch Lightning

[sto1]

Is there a plan to integrate DirectML into the Pytorch Lightning project?
This could broaden the base for DirectML fast.

[Adele101]

Hi @sto1,

Thank you for the great request, we will consider this for an upcoming release of pytorch-directml, and will update this thread when more information is available.

[wuzhican]

is there any update?

[arrrmin]

I would also welcome an update.

[trupljan]

I started implementing a prototype of the integration as a hobby project: Lightning-AI/pytorch-lightning#18188

Current prototype status:
Lightning-AI/pytorch-lightning@master...trupljan:lightning:master

[zhiyuan1i]

Maybe it can be down as a device plugin, I opened a new pr related to this: Lightning-AI/pytorch-lightning#20349