PJRT Örnekleri

Örnek: JAX CUDA eklentisi

  1. Sarmalayıcı aracılığıyla PJRT C API uygulaması (pjrt_c_api_gpu.h).
  2. Paketin giriş noktasını (setup.py) ayarlayın.
  3. initialize() yöntemini (__init__.py) uygulayın.
  4. CUDA için herhangi bir jax testiyle test edilebilir.

Çerçeve Uygulamaları

PJRT cihazlarıyla arayüz oluşturmak için PJRT'yi çerçeve tarafında kullanmayla ilgili bazı referanslar:

Donanım Uygulamaları