"Kullanımı kolay olanı hızlı hale getirip makine öğrenimine uygulamak"
- Yalnızca Python ve Numpy'nin birleşimi
→ XLA kullanarak Numpy'yi GPU/TPU üzerinde derleyip çalıştırır
→ Python fonksiyonlarını tek bir API ile JIT derleyerek XLA ile optimize edilmiş çekirdeklere kolayca yerleştirebilir
→ Birden fazla GPU/TPU üzerinde çalıştırmak da kolaydır (vmap, pmap)
- Mevcut Python+Numpy performansını büyük ölçüde aşar
1 yorum
DeepMind, her şeyi Jax tabanında yeniden düzenledi.
https://deepmind.com/blog/article/using-jax-to-accelerate-our-research