tensorflow / custom-opLinks
Guide for building custom op for TensorFlow
☆385Updated 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☆753Updated 5 months ago
- TVM integration into PyTorch☆456Updated 6 years ago
- ☆600Updated 7 years ago
- The Tensor Algebra SuperOptimizer for Deep Learning☆740Updated 3 years ago
- A tensor-aware point-to-point communication primitive for machine learning☆283Updated last month
- Dive into Deep Learning Compiler☆645Updated 3 years ago
- common in-memory tensor structure☆1,154Updated last week
- ☆371Updated 8 years ago
- Python bindings for NVTX☆67Updated 2 years ago
- Collective communications library with various primitives for multi-machine training.☆1,391Updated 3 weeks ago
- Place for meetup slides☆140Updated 5 years ago
- A GPU performance profiling tool for PyTorch models☆510Updated 4 years ago
- FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/☆1,520Updated last week
- ☆422Updated last month
- TensorFlow/TensorRT integration☆743Updated 2 years ago
- Explore the Capabilities of the TensorRT Platform☆263Updated 4 years ago
- Large Model Support in Tensorflow☆202Updated 5 years ago
- tensorflow源码阅读笔记☆192Updated 7 years ago
- Continuous builder and binary build scripts for pytorch☆356Updated 5 months ago
- PyTorch C++ API Documentation☆244Updated this week
- distributed-embeddings is a library for building large embedding based models in Tensorflow 2.☆46Updated 2 years ago
- To make it easy to benchmark AI accelerators☆194Updated 3 years ago
- A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.☆1,006Updated last year
- HugeCTR is a high efficiency GPU framework designed for Click-Through-Rate (CTR) estimating training☆1,045Updated 4 months ago
- A profiling and performance analysis tool for machine learning☆467Updated this week
- Efficient GPU kernels for block-sparse matrix multiplication and convolution☆1,063Updated 2 years ago
- Documentation for StreamExecutor open source proposal☆83Updated 9 years ago
- A CUDNN minimal deep learning training code sample using LeNet.☆269Updated 2 years ago
- ☆392Updated 3 years ago
- Symbolic Expression and Statement Module for new DSLs☆205Updated 5 years ago