tensorflow / custom-opLinks
Guide for building custom op for TensorFlow
☆382Updated 2 years ago
Alternatives and similar repositories for custom-op
Users that are interested in custom-op are comparing it to the libraries listed below
Sorting:
- A performant and modular runtime for TensorFlow☆758Updated this week
- TVM integration into PyTorch☆453Updated 5 years ago
- ☆588Updated 7 years ago
- common in-memory tensor structure☆1,042Updated last month
- The Tensor Algebra SuperOptimizer for Deep Learning☆726Updated 2 years ago
- A tensor-aware point-to-point communication primitive for machine learning☆259Updated 2 years ago
- Dive into Deep Learning Compiler☆646Updated 3 years ago
- ☆371Updated 7 years ago
- ☆420Updated last week
- A CUDNN minimal deep learning training code sample using LeNet.☆268Updated 2 years ago
- Place for meetup slides☆141Updated 4 years ago
- FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/☆1,415Updated this week
- Documentation for StreamExecutor open source proposal☆83Updated 9 years ago
- distributed-embeddings is a library for building large embedding based models in Tensorflow 2.☆44Updated last year
- Python bindings for NVTX☆66Updated 2 years ago
- A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.☆991Updated 10 months ago
- Explore the Capabilities of the TensorRT Platform☆264Updated 3 years ago
- tensorflow源码阅读笔记☆193Updated 6 years ago
- Symbolic Expression and Statement Module for new DSLs☆205Updated 4 years ago
- TensorFlow/TensorRT integration☆743Updated last year
- Collective communications library with various primitives for multi-machine training.☆1,333Updated last week
- HugeCTR is a high efficiency GPU framework designed for Click-Through-Rate (CTR) estimating training☆1,022Updated 4 months ago
- A profiling and performance analysis tool for machine learning☆405Updated this week
- Computation using data flow graphs for scalable machine learning☆68Updated this week
- PyTorch C++ API Documentation☆232Updated this week
- A GPU performance profiling tool for PyTorch models☆503Updated 4 years ago
- DeepLearning Framework Performance Profiling Toolkit☆285Updated 3 years ago
- Large Model Support in Tensorflow☆202Updated 4 years ago
- Continuous builder and binary build scripts for pytorch☆353Updated 3 months ago
- Efficient Top-K implementation on the GPU☆183Updated 6 years ago