Skip to content

Handle symbolic dimensions in pydantic serialization of vmap_axes. #1179

Handle symbolic dimensions in pydantic serialization of vmap_axes.

Handle symbolic dimensions in pydantic serialization of vmap_axes. #1179