roastduck/FreeTensor
A language and compiler for irregular tensor programs.
Compiles Python-defined loop-based tensor programs to optimized native code while supporting dynamic shapes and explicit user-directed transformations (parallelization, loop restructuring, memory optimization). Generates CUDA or OpenMP-parallelized code and provides reverse-mode automatic differentiation. The compiler leverages an AST-based IR with visitor/mutator patterns for both user-specified schedules and automatic optimization passes.
152 stars. No commits in the last 6 months.
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
symforce-org/symforce
Fast symbolic computation, code generation, and nonlinear optimization for robotics
inducer/loopy
A code generator for array-based code on CPUs and GPUs
alphaville/optimization-engine
Nonconvex embedded optimization: code generation for fast real-time optimization + ROS support
inducer/islpy
Python wrapper for isl, an integer set library
loco-3d/crocoddyl
Crocoddyl is an optimal control library for robot control under contact sequence. Its solver is...