Create your own GitHub profile
Sign up for your own profile on GitHub, the best place to host code, manage projects, and build software alongside 50 million developers.Sign up
42 contributions in the last year
Created a pull request in google/flax that received 4 comments
Created an issue in google/jax that received 17 comments
This is a bug in JAX's shape-checking rules; please report it!.
The function that creates this error is the following: