A domain specific language to express machine learning workloads.
☆1,768Apr 28, 2023Updated 3 years ago
Alternatives and similar repositories for TensorComprehensions
Users that are interested in TensorComprehensions are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Compiler for Neural Network hardware accelerators☆3,318May 11, 2024Updated 2 years ago
- "Multi-Level Intermediate Representation" Compiler Infrastructure☆1,769Apr 22, 2021Updated 5 years ago
- a language for fast, portable data-parallel computation☆6,635Updated this week
- ☆1,651Sep 11, 2018Updated 8 years ago
- Source-to-Source Debuggable Derivatives in Pure Python☆2,325Sep 29, 2022Updated 4 years ago
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- Open Machine Learning Compiler Framework☆13,890Updated this week
- ATen: A TENsor library for C++11☆722Nov 20, 2019Updated 6 years ago
- nGraph has moved to OpenVINO☆1,344Oct 15, 2020Updated 6 years ago
- Write PyTorch code at the level of individual examples, then run it efficiently on minibatches.☆487Feb 12, 2022Updated 4 years ago
- The Tensor Algebra Compiler (taco) computes sparse tensor expressions on CPUs and GPUs☆1,370Apr 14, 2025Updated last year
- Deep universal probabilistic programming with Python and PyTorch☆9,073Oct 5, 2026Updated last month
- Caffe2 is a lightweight, modular, and scalable deep learning framework.☆8,369Feb 7, 2023Updated 3 years ago
- Symbolic Expression and Statement Module for new DSLs☆206Oct 6, 2020Updated 6 years ago
- Tool for real-time visualization, monitoring and collaborative analysis of AI/ML experiments and live data. Supports Python, PyTorch/Torc…☆10,334Updated this week
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.☆14,674Sep 17, 2026Updated last month
- Make huge neural nets fit in memory☆2,843Apr 26, 2020Updated 6 years ago
- Quantized Neural Network PACKage - mobile-optimized implementation of quantized neural network operators☆1,552Aug 28, 2019Updated 7 years ago
- Training Very Deep Neural Networks Without Skip-Connections☆591Jun 9, 2018Updated 8 years ago
- Open standard for machine learning interoperability☆21,656Updated this week
- Collective communications library with various primitives for multi-machine training.☆1,464Oct 1, 2026Updated last month
- FAIR's research platform for object detection research, implementing popular algorithms like Mask R-CNN and RetinaNet.☆26,340Nov 20, 2023Updated 2 years ago
- TensorFlow-based neural network library☆9,974Updated this week
- Learning embeddings for classification, retrieval and ranking.☆3,954Dec 4, 2022Updated 3 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆8,001Jul 14, 2026Updated 3 months ago
- The Tensor Algebra SuperOptimizer for Deep Learning☆744Jan 26, 2023Updated 3 years ago
- Implementations of ideas from recent papers☆390Dec 22, 2020Updated 5 years ago
- Efficiently computes derivatives of NumPy code.☆7,552Updated this week
- common in-memory tensor structure☆1,261Updated this week
- DyNet: The Dynamic Neural Network Toolkit☆3,430Dec 1, 2023Updated 2 years ago
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆9,000Updated this week
- Acceleration package for neural networks on multi-core CPUs☆1,711Jun 11, 2024Updated 2 years ago
- Efficient GPU kernels for block-sparse matrix multiplication and convolution☆1,069Jun 8, 2023Updated 3 years ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- A language-agnostic interface to TensorBoard☆781Nov 9, 2017Updated 8 years ago
- A Python toolbox for performing gradient-free optimization☆4,224Updated this week
- High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.☆4,799Updated this week
- A code generator for array-based code on CPUs and GPUs☆639Oct 28, 2026Updated last week
- Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler; for Python, R, Juli…☆20,793Oct 25, 2023Updated 3 years ago
- PyTorch implementation of "Efficient Neural Architecture Search via Parameters Sharing"☆2,722Jul 6, 2023Updated 3 years ago
- Low-precision matrix multiplication☆1,846Jan 29, 2024Updated 2 years ago