pytorch

Tensor dan jaringan neural dinamis dalam Python dengan akelerasi GPU yang kuat.

Kumpulan contoh seputar PyTorch dalam Visi, Teks, Reinforcement Learning, dll.

Dataset, transformasi, dan model khusus untuk Computer Vision

Tutorial PyTorch.

Platform asli PyTorch untuk melatih model AI generatif.

AI on-device untuk seluler, embedded, dan edge menggunakan PyTorch

Library tingkat tinggi untuk membantu pelatihan dan evaluasi neural network di PyTorch secara fleksibel dan transparan.

Library PyTorch yang modular, mengutamakan primitif, dan berbasis Python untuk Reinforcement Learning.

Kompiler PyTorch/TorchScript/FX untuk GPU NVIDIA menggunakan TensorRT

Kuantisasi native PyTorch untuk pelatihan dan inferensi

Manipulasi dan transformasi data untuk pemrosesan sinyal audio, didukung oleh PyTorch

Mengaktifkan PyTorch pada perangkat XLA (misalnya Google TPU)

FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/

Perpustakaan komunikasi kolektif dengan berbagai primitif untuk pelatihan multi-mesin.

Pengiriman ke https://pytorch.org/hub/

Ekstensi C++ di PyTorch

Pustaka informasi CPU (x86/x86-64/ARM/ARM64, Linux/Windows/Android/macOS/iOS)

TorchBench adalah kumpulan tolok ukur sumber terbuka yang digunakan untuk mengevaluasi kinerja PyTorch.

TensorDict adalah kontainer tensor khusus untuk PyTorch.

Pustaka Profiling CPU+GPU yang menyediakan akses ke jejak lini masa dan penghitung kinerja perangkat keras.

DSL yang tertanam dalam Python untuk memudahkan penulisan kernel ML yang cepat dan terukur dengan boilerplate minimal.