tensorflow / custom-op
Guide for building custom op for TensorFlow
☆378Updated last year
Alternatives and similar repositories for custom-op:
Users that are interested in custom-op are comparing it to the libraries listed below
- A performant and modular runtime for TensorFlow☆759Updated last month
- common in-memory tensor structure☆930Updated 3 months ago
- TVM integration into PyTorch☆453Updated 5 years ago
- A tensor-aware point-to-point communication primitive for machine learning☆252Updated 2 years ago
- Explore the Capabilities of the TensorRT Platform☆261Updated 3 years ago
- ☆402Updated this week
- Python bindings for NVTX☆66Updated last year
- The Tensor Algebra SuperOptimizer for Deep Learning☆696Updated last year
- ☆573Updated 6 years ago
- Dive into Deep Learning Compiler☆646Updated 2 years ago
- A profiling and performance analysis tool for TensorFlow☆363Updated this week
- TensorFlow/TensorRT integration☆738Updated last year
- Dataset, streaming, and file system extensions maintained by TensorFlow SIG-IO☆715Updated this week
- A GPU performance profiling tool for PyTorch models☆500Updated 3 years ago
- ☆375Updated 7 years ago
- DAWNBench: An End-to-End Deep Learning Benchmark and Competition☆261Updated 4 years ago
- [Deprecated] The TensorFlow Profiler (TFProf) UI provides a visual interface for profiling TensorFlow models.☆136Updated 5 years ago
- TensorFlow-nGraph bridge☆137Updated 3 years ago
- FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/☆1,240Updated this week
- tensorflow源码阅读笔记☆190Updated 6 years ago
- distributed-embeddings is a library for building large embedding based models in Tensorflow 2.☆43Updated last year
- TensorFlow examples in C, C++, Go and Python without bazel but with cmake and FindTensorFlow.cmake☆439Updated 5 years ago
- Place for meetup slides☆140Updated 4 years ago
- TensorFlow Estimator☆304Updated 11 months ago
- Example repository for custom C++/CUDA operators for TorchScript☆114Updated 2 years ago
- To make it easy to benchmark AI accelerators☆182Updated 2 years ago
- A simple memory manager for CUDA designed to help Deep Learning frameworks manage memory☆296Updated 6 years ago
- A CUDNN minimal deep learning training code sample using LeNet.☆265Updated last year
- Running BERT without Padding☆468Updated 2 years ago
- Collective communications library with various primitives for multi-machine training.☆1,253Updated 2 weeks ago