Evan Quiney 1200a7db64 Add tensor sharding for GPT-OSS (#1144) ...

## Motivation

GPT OSS did not previously support tensor sharding

## Changes

Add GPT sharding support in tensor_auto_parallel.
Code is mostly @rltakashige's

## Test Plan

### Manual Testing
Tested GPT-OSS - MLX Fast Sync causes issues in Tensor RDMA - this is a general problem at the moment.

2026-01-13 17:25:52 +00:00

..

headless_runner.py

Add tensor sharding for GPT-OSS (#1144)

2026-01-13 17:25:52 +00:00

start_distributed_test.sh

Upgrade mlx-lm to 0.30.2 with transformers 5.x compatibility (#1125)

2026-01-13 12:06:04 +00:00