Raj Thakur
Raj Thakur
## 🐛 Bug `all_to_all` operation generates invalid HLO that fails verification with error: `RET_CHECK failure hlo->operand_count() == split_count`. The generated HLO `all-to-all` instruction is missing required attributes (`split_dimension`, `concat_dimension`, `split_count`)...
## 🐛 Bug We are observing a memory leak issue as PJRT_Client_Destroy is not invoked in PT/XLA 2.8.0+ causing backend runtime to be not properly cleaned for AWS neuron devices....
## 🐛 Bug The maybe_get_jax() function in torch_xla/_internal/jax_workarounds.py merged in #9521 currently emits a warning message when JAX is not installed. While informative, this warning results in an excessive number...