Brandon Monsen issues

Repositories
Issues
Comments

Results 1 issues of


                                            Brandon Monsen

Jax grad doesn't work with tfp jax module's Gram-Schmidt function because of a lax for loop

### Issue type Support ### Have you reproduced the bug with TensorFlow Nightly? Yes ### Source source ### TensorFlow version 0.24.0.dev20240121 ### Custom code Yes ### OS platform and distribution...

stat:awaiting response

type:support

stale