jax
5c1de283 - revise vmap in_axes/out_axes leaf type error msg (#3179)

Commit

5 years ago

revise vmap in_axes/out_axes leaf type error msg (#3179) from #3161 discussion

References

#3179 - revise vmap in_axes/out_axes leaf type error msg

Author

mattjj

mattjj

Parents

Loading