3 ms·
This is not comparable with ML frameworks (Pytorch or Jax) - its closest analogue is Halide (which they mention in the paper).
by chillee 5y ago
This is not comparable with ML frameworks (Pytorch or Jax) - its closest analogue is Halide (which they mention in the paper).
- nestorD 5y agoI would say it is somewhat comparable with XLA which is the compiler backend for Jax (and was heavily inspired by Halide).
- brrrrrm 5y agoI'd be curious to read up on the parts of XLA design that were inspired by Halide if you have any links. I haven't familiarized myself with XLA/MLIR source code much, but I was under the impression it took a more conventional approach to optimizing compilers (pattern matching rather than generic separation of schedule and compute).