Distributed# Helper Routines# The mlx.nn.layers.distributed package contains helpful routines to create sharded layers from existing Modules. shard_linear shard_inplace Layers# AllToShardedLinear ShardedToAllLinear QuantizedAllToShardedLinear QuantizedShardedToAllLinear
