microsoft / tensorflow-directml-plugin
DirectML PluggableDevice plugin for TensorFlow 2
☆192Updated 7 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
- Fork of TensorFlow accelerated by DirectML☆461Updated 3 months ago
- DirectML is a high-performance, hardware-accelerated DirectX 12 library for machine learning. DirectML provides GPU acceleration for comm…☆2,317Updated 2 months ago
- Intel® Extension for TensorFlow*☆329Updated last week
- ONNX Script enables developers to naturally author ONNX functions and models using a subset of Python.☆305Updated this week
- The TensorFlow-specific implementation of the Keras API, which was the default Keras from 2019 to 2023.☆66Updated this week
- ONNX Runtime: cross-platform, high performance scoring engine for ML models☆58Updated this week
- 🤗 Optimum Intel: Accelerate inference with Intel optimization tools☆431Updated this week
- ONNXMLTools enables conversion of models to ONNX☆1,042Updated last week
- Antares: an automatic engine for multi-platform kernel generation and optimization. Supporting CPU, CUDA, ROCm, DirectX12, GraphCore, SYC…☆452Updated last week
- onnxruntime-extensions: A specialized pre- and post- processing library for ONNX Runtime☆350Updated this week
- Examples for using ONNX Runtime for model training.☆323Updated 2 months ago
- Common utilities for ONNX converters☆256Updated last month
- ☆85Updated last year
- Tensors and Dynamic neural networks in Python with strong GPU acceleration☆220Updated this week
- AMD's graph optimization engine.☆196Updated this week
- MIVisionX toolkit is a set of comprehensive computer vision and machine intelligence libraries, utilities, and applications bundled into …☆190Updated last week
- cudnn_frontend provides a c++ wrapper for the cudnn backend API and samples on how to use it☆484Updated last month
- Intel® NPU Acceleration Library☆582Updated last week
- ☆100Updated 2 months ago
- A collection of examples for the ROCm software stack☆178Updated this week
- Generative AI extensions for onnxruntime☆585Updated this week
- Convert scikit-learn models and pipelines to ONNX☆564Updated last month
- Data Parallel Extension for Numba☆78Updated 2 months ago
- An efficient C++17 GPU numerical computing library with Python-like syntax☆1,241Updated this week
- C++ API for ML inferencing and transfer-learning on Coral devices☆87Updated 5 months ago
- ☆449Updated last month
- CUDA Python: Performance meets Productivity☆1,047Updated this week
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆292Updated this week
- oneAPI Math Library (oneMath)☆636Updated last week
- Backward compatible ML compute opset inspired by HLO/MHLO☆428Updated this week