Penzai - Sinir ağları oluşturma, düzenleme ve görselleştirme için JAX araç kiti

• Google DeepMind tarafından yayımlanan, modelleri okunması kolay ve işlevsel Pytree veri yapılarıyla yazmak için bir JAX kütüphanesi
• Model eğitildikten sonra bile üzerinde kolayca çalışılabilmesine odaklanır
  • Model bileşenlerini tersine mühendislik etme veya kaldırma; iç aktivasyonları inceleme ve araştırma; model cerrahisi; mimari hata ayıklama gibi araştırmalar için uygundur
  • Model görselleştirme, değiştirme ve analiz araçları sunar; bu sayede yalnızca model oluşturma ve eğitme amacıyla da kullanılabilir
• Birlikte tasarlanmış, ancak her biri bağımsız olarak da kullanılabilen modüler bir araç setinden oluşur

Dahil olan araçlar

penzai.nn (pz.nn)

• Flax, Haiku, Keras, Equinox gibi diğer sinir ağı kütüphanelerine alternatif olan, deklaratif kombinatör tabanlı bir sinir ağı kütüphanesi
• Modelin forward pass yapısının tamamını model pytree'sine açığa çıkarır; böylece pretty printing ile modelin tüm işlemlerini görebilir ve jax.tree_util ile yeni çalışma zamanı mantığı enjekte edebilirsiniz
• Model, dahili olarak yalnızca çağrılabilir bir pytree'dir

penzai.treescope (pz.ts)

• Standart IPython/Colab renderer'ının yerine drop-in şekilde çalışan güçlü ve etkileşimli bir Python pretty-printer
• Penzai modellerini ve derin biçimde iç içe geçmiş diğer JAX pytree'lerini anlamaya yardımcı olmak için tasarlanmıştır ve keyfi boyutlardaki NDArray görselleştirmesini yerleşik olarak destekler

penzai.core.selectors (pz.select)

• JAX'in .at[...].set(...) sözdizimini, rastgele tür tabanlı pytree gezinimine genelleştiren çok amaçlı bir pytree aracı
• Penzai modellerinde ve diğer veri yapılarında karmaşık yeniden yazımları veya anlık patch uygulamalarını kolayca gerçekleştirebilir

penzai.core.named_axes (pz.nx)

• Sıradan JAX fonksiyonlarını adlandırılmış eksenler üzerinde vektörleştirebilen ve yeni bir dizi API'si öğrenmeden adlandırılmış programlama stili ile konumsal programlama stili arasında sorunsuz geçiş sağlayan hafif bir adlandırılmış eksen sistemi

penzai.data_effects (pz.de)

• Pytree gezinimi üzerine kurulu; model yazımını veya kullanımını engellemeden ek argümanlar, rastgelelik ve durum değişkenleri üzerinde kontrol sağlayan opt-in bir sistem

[brainer]

Pytorch için benzer bir şey var mı acaba?