microsoft / tensorflow-directml-pluginLinks
DirectML PluggableDevice plugin for TensorFlow 2
☆192Updated 3 months ago
Alternatives and similar repositories for tensorflow-directml-plugin
Users that are interested in tensorflow-directml-plugin are comparing it to the libraries listed below
Sorting:
- Fork of TensorFlow accelerated by DirectML☆467Updated 8 months ago
- DirectML is a high-performance, hardware-accelerated DirectX 12 library for machine learning. DirectML provides GPU acceleration for comm…☆2,460Updated last month
- ONNX Script enables developers to naturally author ONNX functions and models using a subset of Python.☆354Updated this week
- Intel® Extension for TensorFlow*☆337Updated 2 months ago
- 🤗 Optimum Intel: Accelerate inference with Intel optimization tools☆466Updated this week
- ☆106Updated last month
- Tensors and Dynamic neural networks in Python with strong GPU acceleration☆227Updated this week
- AMD's graph optimization engine.☆220Updated this week
- onnxruntime-extensions: A specialized pre- and post- processing library for ONNX Runtime☆388Updated last week
- ☆47Updated this week
- ☆86Updated last year
- Examples for using ONNX Runtime for model training.☆338Updated 7 months ago
- A profiling and performance analysis tool for machine learning☆386Updated this week
- Python SYCL bindings and SYCL-based Python Array API library☆112Updated last week
- ONNX Optimizer☆715Updated this week
- BLAS-like Library Instantiation Software Framework☆141Updated this week
- cudnn_frontend provides a c++ wrapper for the cudnn backend API and samples on how to use it☆568Updated this week
- The NVIDIA® Tools Extension SDK (NVTX) is a C-based Application Programming Interface (API) for annotating events, code ranges, and resou…☆385Updated this week
- Computation using data flow graphs for scalable machine learning☆67Updated this week
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆329Updated this week
- Common utilities for ONNX converters☆270Updated 5 months ago
- oneAPI Specification source files☆201Updated this week
- Python API for ML inferencing and transfer-learning on Coral devices☆386Updated last year
- Olive: Simplify ML Model Finetuning, Conversion, Quantization, and Optimization for CPUs, GPUs and NPUs.☆1,944Updated this week
- Intel® AI Reference Models: contains Intel optimizations for running deep learning workloads on Intel® Xeon® Scalable processors and Inte…☆716Updated last week
- ☆523Updated last week
- Generative AI extensions for onnxruntime☆722Updated this week
- MIVisionX toolkit is a set of comprehensive computer vision and machine intelligence libraries, utilities, and applications bundled into …☆194Updated this week
- A collection of examples for the ROCm software stack☆215Updated this week
- Antares: an automatic engine for multi-platform kernel generation and optimization. Supporting CPU, CUDA, ROCm, DirectX12, GraphCore, SYC…☆470Updated last month