Accelerate expanded hardware support and cleaned up its device abstraction layer. AWS Neuron (Trainium/Inferentia) integration shipped alongside removal of the IPEX dependency, making XPU code more device-agnostic through spawn-based process launching. FSDP2 picked up fixes for gradient upcasting, tied embedding errors, and optimizer state loading.
Accelerate expanded hardware support and cleaned up dependencies in March. AWS Neuron integration arrived for Trainium/Inferentia devices, while IPEX was removed and XPU code was refactored to be device-agnostic. FSDP2 received multiple stability fixes including gradient upcasting improvements, tied embedding error handling, and optimizer state loading.