flax
Plumb spmd_axis_name through transforms.vmap through to JAX vmap
#2398
Merged
Overview
Commits
0
Changes
View On GitHub
FAQ Terms Privacy Refunds Impressum
Loading