pytensor icon indicating copy to clipboard operation
pytensor copied to clipboard

Default typify in JITLinkers expects tensors/arrays, why?

Open ricardoV94 opened this issue 1 year ago • 0 comments

Description

https://github.com/pymc-devs/pytensor/blob/0e29d76b3122ba4fdedf0531005a4c0ba178d270/pytensor/link/jax/dispatch/basic.py#L22-L28

(And now from copying the approach also the PyTorch Linker)

ricardoV94 avatar Jun 14 '24 15:06 ricardoV94