jax-ml/jax — 36k+ yıldız · Python

Ne işe yarar?

GPU/TPU'ya derleyerek hızlandırıcı odaklı dizi hesaplaması hedefler.

Nasıl kullanılır?

jax'ı kur, NumPy tarzı kod yaz ve grad, vmap, jit dönüşümlerini uygula.

Kuzey Kutbu ipucu

Bu araçları keşfederken kodun temelini Kuzey Kutbu dersleriyle sağlam at; yapay zekâyı bir yardımcı gibi kullan, ürettiği kodu anlamadan kopyalama.

GitHub'da aç ↗