Numba?

116 views
Skip to first unread message

Noah Brenowitz

unread,
Jul 21, 2022, 7:08:41 PM7/21/22
to Enzyme AD
Has anyone tried using enzyme on llvm IR generate by numba?

We have an intern accelerating some physics codes with numba, and it would be amazing to get automatic differentiation "for free". numba has some hooks to view the generated IR: https://numba.pydata.org/numba-doc/latest/reference/jit-compilation.html?highlight=llvm#Dispatcher.inspect_llvm.

Best,
Noah
Reply all
Reply to author
Forward
0 new messages