Can TVM compile a given Python function?

Questions
1

Can TVM compile a given Python function using a decorator syntax, like this

@tvm.jit
def f(a, b, c):
  return a @ b + c

does something like this exist? Or in general how do I compile a Python function? There’s no DNN model here, just a Python function. Thanks!

2

TVM is not designed to compile a generic python program. Matxscript, which is built with similar ideas of TVM’s PackedFunc system, broadly expand the territory of what python program it could compile, and perhaps it’s worth a try: https://github.com/bytedance/matxscript

3

That’s very helpful! Thanks for the suggestions!