DirectML Plugin for TensorFlow 2 is here

Last October we announced our generally consumable DirectML backend for TensorFlow 1.15. Since then, our team worked to bring machine learning training acceleration to TensorFlow 2. Today, we are happy to announce the release of our DirectML PluggableDevice preview package for TensorFlow on PyPI. Alongside our release, we are excited to ...

PyTorch-DirectML: Preview Release 3

The Windows AI Platform is excited to announce the Third Preview release of PyTorch-DirectML; a hardware accelerated backend for training PyTorch models on any DirectX12 GPU on Windows and the Windows Subsystem for Linux (WSL). This preview release includes expanded support for computer vision models from the Torchvision library. On the ...

PyTorch-DirectML: Preview Release 2

A few months ago, we released the first preview of PyTorch-DirectML: a hardware accelerated backend for training PyTorch models on any DirectX12 GPU on Windows and the Windows Subsystem for Linux (WSL). Today, we are releasing the Second Preview with significant performance improvements and greater coverage for computer vision models. ...
Comments are closed.0

Introducing PyTorch-DirectML: Train your machine learning models on any GPU

The Windows AI team is excited to announce the first preview of DirectML as a backend to PyTorch for training ML models! This release is our first step towards unlocking accelerated machine learning training for PyTorch on any DirectX12 GPU on Windows and the Windows Subsystem for Linux (WSL). In order for you to take advantage of DirectML ...

Train your machine learning models on any GPU with TensorFlow-DirectML

Learn about the first generally consumable package of TensorFlow-DirectML. TensorFlow-DirectML improves the experience and performance of model training through GPU acceleration on the breadth of Windows devices by working across different hardware vendors and supporting the needs of students’ and professionals’ workflows.