xla
Add support for callable in torchax.interop.JittableModule.functional_call in the first parameter
#9451
Merged
Overview
Commits
1
Changes
View On GitHub

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

qihqi merged 1 commit into pytorch:master from zmelumian972:torchax/jittable_module_callable
zmelumian972
qihqi
qihqi approved these changes on 2025-07-08
zmelumian972
qihqi
zmelumian JittableModule can now work with callable method instead of just meth…
f1f6894b
zmelumian972 zmelumian972 force pushed from dc21e019 to f1f6894b 163 days ago
zmelumian972
qihqi qihqi merged ca914452 into master 158 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum