microsoft / DirectML
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.
☆2,333Updated 2 months ago
Alternatives and similar repositories for DirectML:
Users that are interested in DirectML are comparing it to the libraries listed below
- Fork of TensorFlow accelerated by DirectML☆462Updated 4 months ago
- A Python package for extending the official PyTorch that can easily obtain performance on Intel platform☆1,716Updated this week
- DirectML PluggableDevice plugin for TensorFlow 2☆192Updated 8 months ago
- Olive: Simplify ML Model Finetuning, Conversion, Quantization, and Optimization for CPUs, GPUs and NPUs.☆1,728Updated this week
- AMD ROCm™ Software - GitHub Home☆4,870Updated this week
- Simple, safe way to store and distribute tensors☆3,026Updated 2 weeks ago
- A machine learning compiler for GPUs, CPUs, and ML accelerators☆2,874Updated this week
- Examples for using ONNX Runtime for machine learning inferencing.☆1,276Updated last week
- High-efficiency floating-point neural network inference operators for mobile, server, and Web☆1,929Updated this week
- TensorFlow ROCm port☆690Updated this week
- Generative AI extensions for onnxruntime☆591Updated this week
- AMD's Machine Intelligence Library☆1,109Updated this week
- Examples for using ONNX Runtime for model training.☆324Updated 3 months ago
- Intel® NPU Acceleration Library☆593Updated 2 weeks ago
- ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator☆15,415Updated this week
- oneAPI Deep Neural Network Library (oneDNN)☆3,698Updated this week
- General purpose GPU compute framework built on Vulkan to support 1000s of cross vendor graphics cards (AMD, Qualcomm, NVIDIA & friends). …☆2,074Updated last month
- 🚀 Accelerate inference and training of 🤗 Transformers, Diffusers, TIMM and Sentence Transformers with easy to use hardware optimization…☆2,695Updated this week
- Intel® Extension for TensorFlow*☆329Updated 2 weeks ago
- Actively maintained ONNX Optimizer☆662Updated this week
- CUDA Templates for Linear Algebra Subroutines☆6,076Updated this week
- ONNXMLTools enables conversion of models to ONNX☆1,042Updated 3 weeks ago
- SHARK Studio -- Web UI for SHARK+IREE High Performance Machine Learning Distribution☆1,438Updated 3 months ago
- cudnn_frontend provides a c++ wrapper for the cudnn backend API and samples on how to use it☆488Updated this week
- A retargetable MLIR-based machine learning compiler and runtime toolkit.☆2,948Updated this week
- 🤗 Optimum Intel: Accelerate inference with Intel optimization tools☆436Updated this week
- PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT☆2,658Updated this week
- Tensors and Dynamic neural networks in Python with strong GPU acceleration☆220Updated this week
- Convert TensorFlow, Keras, Tensorflow.js and Tflite models to ONNX☆2,360Updated 2 weeks ago
- ⚡ Build your chatbot within minutes on your favorite device; offer SOTA compression techniques for LLMs; run LLMs efficiently on Intel Pl…☆2,155Updated 3 months ago