microsoft / tensorflow-directml-plugin
DirectML PluggableDevice plugin for TensorFlow 2
☆193Updated 2 weeks 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☆464Updated 5 months ago
- 🤗 Optimum Intel: Accelerate inference with Intel optimization tools☆447Updated this week
- Intel® Extension for TensorFlow*☆332Updated last week
- DirectML is a high-performance, hardware-accelerated DirectX 12 library for machine learning. DirectML provides GPU acceleration for comm…☆2,382Updated 3 months ago
- Python SYCL bindings and SYCL-based Python Array API library☆110Updated this week
- The TensorFlow-specific implementation of the Keras API, which was the default Keras from 2019 to 2023.☆74Updated last week
- ☆105Updated last week
- Run Generative AI models with simple C++/Python API and using OpenVINO Runtime☆227Updated this week
- scikit-learn_bench benchmarks various implementations of machine learning algorithms across data analytics frameworks. It currently suppo…☆115Updated last week
- cudnn_frontend provides a c++ wrapper for the cudnn backend API and samples on how to use it☆515Updated last month
- Generative AI extensions for onnxruntime☆645Updated this week
- Repository for OpenVINO's extra modules☆116Updated last month
- OpenVINO Tokenizers extension☆31Updated this week
- ONNX Script enables developers to naturally author ONNX functions and models using a subset of Python.☆323Updated this week
- A collection of examples for the ROCm software stack☆190Updated this week
- ONNX Runtime: cross-platform, high performance scoring engine for ML models☆59Updated this week
- Data Parallel Extension for Numba☆79Updated 3 months ago
- AMD's graph optimization engine.☆213Updated this week
- Examples for using ONNX Runtime for model training.☆329Updated 4 months ago
- ONNXMLTools enables conversion of models to ONNX☆1,055Updated 2 months ago
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆313Updated this week
- Intel® NPU Acceleration Library☆642Updated 2 months ago
- onnxruntime-extensions: A specialized pre- and post- processing library for ONNX Runtime☆365Updated this week
- ☆85Updated last year
- The Foundation for All Legate Libraries☆205Updated last week
- LLM SDK for OnnxRuntime GenAI (OGA)☆104Updated this week
- AMD's Machine Intelligence Library☆1,125Updated this week
- Intel® Extension for DeepSpeed* is an extension to DeepSpeed that brings feature support with SYCL kernels on Intel GPU(XPU) device. Note…☆61Updated last week
- OpenAI Triton backend for Intel® GPUs☆168Updated this week
- A Python-level JIT compiler designed to make unmodified PyTorch programs faster.☆1,035Updated 10 months ago