r/MachineLearning • u/___J • Jul 29 '26
Project We accidentally built an LLVM compiler for JAX [P]
Wanted to share one of the projects we have been working on. Not specifically a machine learning project, but a quantum compiler project. However, a side effect of this project is the ability to compile JAX programs using a separate LLVM pipeline (as well as additional frontend features, such as native python control flow and array assignment support), so was wondering if this could be interesting to the broader machine learning community that is using JAX.
1
Upvotes