xla
Add support for callable in torchax.interop.JittableModule.functional_call in the first parameter
#9451
Merged
Merged
Add support for callable in torchax.interop.JittableModule.functional_call in the first parameter #9451
qihqi
merged
1 commit into
pytorch:master
from
torchax/jittable_module_callable
qihqi
approved these changes
on 2025-07-08
JittableModule can now work with callable method instead of just meth…
f1f6894b
Login to write a write a comment.
Login via GitHub