jax
107689e9 - improve vmap axis spec structure mismatch errors (#3619)

Commit

6 years ago

improve vmap axis spec structure mismatch errors (#3619) * improve vmap axis spec structure mismatch errors fixes #3613 * deflake

References

Author

mattjj

Parents