partition spec

Concept

Used with a mesh in Jax to map tensors to the physical hardware grid, defining how data is sharded or replicated across dimensions.

Mentioned in 1 video

Videos Mentioning partition spec

AI Dev 25 x NYC | Robert Crowe: JAX Made Simple: An Intuitive Guide to Building Fast Neural Networks

AI Dev 25 x NYC | Robert Crowe: JAX Made Simple: An Intuitive Guide to Building Fast Neural Networks

DeepLearningAI

Used with a mesh in Jax to map tensors to the physical hardware grid, defining how data is sharded or replicated across dimensions.