PJRT উদাহরণ

উদাহরণ: JAX CUDA প্লাগইন

  1. মোড়কের মাধ্যমে PJRT C API বাস্তবায়ন ( pjrt_c_api_gpu.h )।
  2. প্যাকেজের জন্য এন্ট্রি পয়েন্ট সেট আপ করুন ( setup.py )।
  3. একটি ইনিশিয়ালাইজ() পদ্ধতি প্রয়োগ করুন ( __init__.py )।
  4. CUDA-এর জন্য যেকোন জ্যাক্স পরীক্ষা দিয়ে পরীক্ষা করা যেতে পারে।

ফ্রেমওয়ার্ক বাস্তবায়ন

PJRT ডিভাইসের সাথে ইন্টারফেস করতে ফ্রেমওয়ার্ক সাইডে PJRT ব্যবহার করার জন্য কিছু রেফারেন্স:

  • JAX
    • jax-ml/jax xla_client API-এর মাধ্যমে PJRT API-এর সাথে ইন্টারঅ্যাক্ট করে
  • GoMLX
  • জেডএমএল
    • PJRT API মোড়ক pjrt.zig
    • PJRT প্লাগইন context.zig লোড করুন
    • PJRT Buffers buffer.zig এর সাথে ইন্টারঅ্যাক্ট করা
    • PJRT module.zig এর মাধ্যমে একটি মডিউল চালান

হার্ডওয়্যার বাস্তবায়ন