jax
bdcd358b - improve while_loop carry pytree/type mismatch errors

Commit

1 year ago

improve while_loop carry pytree/type mismatch errors Now we call into the same error utility as we use in scan.

References

Author

mattjj

Committer

mattjj

Parents