tensorflow / custom-op
Guide for building custom op for TensorFlow
☆381Updated 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
- A performant and modular runtime for TensorFlow☆759Updated last week
- TVM integration into PyTorch☆452Updated 5 years ago
- Python bindings for NVTX☆66Updated last year
- ☆410Updated last week
- A tensor-aware point-to-point communication primitive for machine learning☆257Updated 2 years ago
- ☆582Updated 7 years ago
- ☆372Updated 7 years ago
- Explore the Capabilities of the TensorRT Platform☆264Updated 3 years ago
- Place for meetup slides☆140Updated 4 years ago
- TensorFlow/TensorRT integration☆741Updated last year
- Large Model Support in Tensorflow☆202Updated 4 years ago
- A profiling and performance analysis tool for machine learning☆372Updated this week
- Dive into Deep Learning Compiler☆646Updated 2 years ago
- common in-memory tensor structure☆982Updated 2 weeks ago
- Running BERT without Padding☆471Updated 3 years ago
- distributed-embeddings is a library for building large embedding based models in Tensorflow 2.☆44Updated last year
- Symbolic Expression and Statement Module for new DSLs☆205Updated 4 years ago
- TensorFlow examples in C, C++, Go and Python without bazel but with cmake and FindTensorFlow.cmake☆439Updated 5 years ago
- The Tensor Algebra SuperOptimizer for Deep Learning☆707Updated 2 years ago
- HierarchicalKV is a part of NVIDIA Merlin and provides hierarchical key-value storage to meet RecSys requirements. The key capability of…☆143Updated this week
- Enhanced networking support for TensorFlow. Maintained by SIG-networking.☆98Updated 3 years ago
- tensorflow源码阅读笔记☆190Updated 6 years ago
- TensorFlow-nGraph bridge☆136Updated 4 years ago
- Dataset, streaming, and file system extensions maintained by TensorFlow SIG-IO☆724Updated this week
- Computation using data flow graphs for scalable machine learning☆67Updated this week
- A GPU performance profiling tool for PyTorch models☆505Updated 3 years ago
- Convert nvprof profiles into about:tracing compatible JSON files☆69Updated 4 years ago
- CS294; AI For Systems and Systems For AI☆224Updated 5 years ago
- [Deprecated] The TensorFlow Profiler (TFProf) UI provides a visual interface for profiling TensorFlow models.☆136Updated 5 years ago
- A simple memory manager for CUDA designed to help Deep Learning frameworks manage memory☆297Updated 6 years ago