roastduck/FreeTensor
A language and compiler for irregular tensor programs.
This tool helps developers write high-performance numerical programs in Python by compiling their loop-based tensor code into optimized native code. It takes Python code defining tensor operations and outputs highly efficient executables, making it useful for machine learning engineers, scientific programmers, or anyone needing to accelerate computational tasks.
152 stars. No commits in the last 6 months.
Use this if you need to significantly speed up your Python-based tensor computations, especially those with dynamic shapes or irregular access patterns, by compiling them to native code and applying explicit optimizations like parallelization.
Not ideal if you primarily work with existing high-level deep learning frameworks like TensorFlow or PyTorch and don't require low-level control over loop transformations and memory management.
Stars
152
Forks
12
Language
C++
License
Apache-2.0
Category
Last pushed
Nov 29, 2024
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ai-coding/roastduck/FreeTensor"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
alphaville/optimization-engine
Nonconvex embedded optimization: code generation for fast real-time optimization + ROS support
symforce-org/symforce
Fast symbolic computation, code generation, and nonlinear optimization for robotics
stack-of-tasks/pinocchio
A fast and flexible implementation of Rigid Body Dynamics algorithms and their analytical derivatives
inducer/loopy
A code generator for array-based code on CPUs and GPUs
loco-3d/crocoddyl
Crocoddyl is an optimal control library for robot control under contact sequence. Its solver is...