xla
Add support for callable in torchax.interop.JittableModule.functional_call in the first parameter
#9451

Merged

Commits

JittableModule can now work with callable method instead of just method name in `functional_call`

zmelumian committed 1 year ago

Loading