JAX'ın Hesaplamayı Ağa Üzerine Parçalama Yöntemleri
JAX, hesaplamayı ağa parçalamak için üç yöntem sunuyor. Bu yazıda, her birinin detaylarını inceliyoruz.
JAX, hesaplama yerleştirmesinde üç ana yöntemi kullanıyor: otomatik, açık ve manuel. Bu yazıda, her bir yöntemi gerçek jaxprs, HLO ve MLIR ile 8 cihazlı bir çalışmadan örneklerle inceleyerek, derleyicinin doğrulukta nasıl bir ortak haline geldiğini gösteriyorum.