Commits

Commits on Oct 17, 2024

Commits on Oct 7, 2024

Part 1: Introduce multi-node SPMD support for Neuron (#8204) (#8224)

jeffhataws
and
rpsilva-aws
authored

Commits on Oct 4, 2024

Commits on Oct 3, 2024

change r2.5 to use jax stable version 0.4.33 (#8062)
ManfeiBai
authored

Commits on Oct 2, 2024

[Cherry-pick] Enable cross entropy loss for xla autocast with FP32 precision (#8094) (#8201)

ManfeiBai
and
avizon-aws
authored

Commits on Sep 27, 2024

[cherry-pick] Update Openxla-pin to Sep13, libtpu-pin to Sep13, jax to 0.4.33 stable version (#8052)
ManfeiBai
authored
Force github workflow catch PyTorch commit from release/2.5 branch for PyTorch/XLA `r2.5` branch (#8078)
ManfeiBai
authored
cherry pick collective op all_to_all_single (#8077)
zpcore
authored

Commits on Sep 25, 2024

Use Tuple instead of tuple in scan.py (#8067)
tengyifei
authored

Commits on Sep 23, 2024

Set xla_tpu_enable_flash_attention=false to enable libtpu pin update (#8008)

bhavya01
authored and
ManfeiBai
committed
Follow 2 releases grace period for API deprecation (#8003)

zpcore
authored and
ManfeiBai
committed

Commits on Sep 9, 2024

Commits on Sep 3, 2024

Skip dynamo collective tests on TPU v2 CI (#7949)
will-cromar
authored