pytensor icon indicating copy to clipboard operation
pytensor copied to clipboard
verified publisher icon pymc-devs

Add itypes/otypes specification for as_jax_op/wrap_jax

Open aseyboldt opened this issue 3 months ago • 0 comments

Description

Users should be able to set ityes and otypes for as_jax_op/wrap_jax directly if they want to avoid automatic inference, just as in as_op.

aseyboldt avatar Sep 24 '25 12:09 aseyboldt