Download
Tensor Comprehensions (TC) is a fully functional C++ library that automatically synthesizes high-performance machine learning kernels using Halide, ISL, and NVRTC or LLVM. TC additionally provides basic integration with Caffe2 and PyTorch. We provide more details in our paper on arXiv. This library is designed to be highly portable, machine-learning-framework agnostic and only requires a simple tensor library with memory allocation, offloading, and synchronization capabilities.
Features
- Documentation available
- Examples available
- This library is designed to be highly portable
- Machine-learning-framework agnostic
- Only requires a simple tensor library with memory allocation, offloading and synchronization capabilities
- Basic integration with Caffe2
Project Samples
Categories
Machine LearningLicense
Apache License V2.0Follow Tensor Comprehensions
Other Useful Business Software
MicroStation by Bentley Systems is the trusted computer-aided design (CAD) software built specifically for infrastructure design.
MicroStation is the only computer-aided design software for infrastructure design, helping architects and engineers like you bring their vision to life, present their designs to their clients, and deliver their projects to the community.
Rate This Project
Login To Rate This Project
User Reviews
Be the first to post a review of Tensor Comprehensions!
