DirectML is a high-performance, hardware-accelerated DirectX 12 library for machine learning. DirectML provides GPU acceleration for common machine learning tasks across a broad range of supported hardware and drivers, including all DirectX 12-capable GPUs from vendors such as AMD, Intel, NVIDIA, and Qualcomm.
MIT License
Bot releases are hidden (Show)
Preview build of torch-directml built on August 15, 2024.
The Python packages are available as a PyPI release. To download the appropriate python package automatically, simply pip install torch-directml
.
avg_pool3d
, upsample_bicubic2d
device_count
in torch PrivateUse1
backenddevice_count
inPrivateUse1
#596 #609Published by joshjkim 3 months ago
Preview build of torch-directml built on July 15, 2024.
The Python packages are available as a PyPI release. To download the appropriate python package automatically, simply pip install torch-directml
.
isin
, std_mean
, group_norm
, multinomial
Published by joshjkim 4 months ago
Preview build of torch-directml built on June 14, 2024.
The Python packages are available as a PyPI release. To download the appropriate python package automatically, simply pip install torch-directml
.
softplus
, amax
, linspace
, vector_norm
, native_dropout
linear
operatorPrivateUse1
backendPrivateUse1
random seed support #592 #590 #587 #586Published by jstoecker over 4 years ago
Preview build of tensorflow-directml built on June 26, 2020.
The Python packages are available as a PyPI release. To download the appropriate python package automatically, simply pip install tensorflow-directml
.
• Certain OOM errors will now gracefully fail instead of crashing the python interpreter.
• Allow usage of entire graphics memory when grappler optimizations are enabled.
• GRUBlockCell operator implemented.
Published by jstoecker over 4 years ago
First preview of TensorFlow 1.15.3 with DirectML support in both Windows and WSL.
Please visit tensorflow-directml on PyPI.org for the latest release!