QuantizedAllToShardedLinear

Quantized distributed linear layer.

Abstract Signature:

QuantizedAllToShardedLinear(in_features: int, out_features: int)

PyTorchApple MLX

PyTorch

API: —

Strategy: Custom / Partial

Apple MLX

API: mlx.nn.QuantizedAllToShardedLinear

Strategy: Direct Mapping

Official Docs ↗