Pytensor function needs at least one output for Jax

Updating PyTensor solved it, thanks

1 Like