OpenXLA عالی

اکوسیستم OpenXLA

OpenXLA یک اکوسیستم باز از اجزای زیرساخت یادگیری ماشین (ML) کارآمد، قابل حمل و توسعه‌پذیر است که توسعه ML را با یکپارچه‌سازی ابزارها بین فریم‌ورک‌های فرانت‌اند و پشتیبان‌های سخت‌افزار ساده می‌کند. ساخته شده توسط رهبران صنعت در مدل سازی، نرم افزار و سخت افزار هوش مصنوعی.

جامعه چگونه از OpenXLA استفاده می کند؟ این صفحه پیوندها را به مخازن و پروژه ها با استفاده از OpenXLA برای ارائه الهام و نشانگرهای کد یکپارچه می کند!

پروژه ای دارید که از OpenXLA استفاده می کند؟ یک درخواست کشش برای ما ارسال کنید و آن را به این صفحه اضافه کنید!

چارچوب ها

  • JAX یک چارچوب ML با یک API مانند NumPy برای نوشتن مدل‌های ML با کارایی بالا است.
  • PyTorch/XLA پلی از PyTorch به OpenXLA و StableHLO فراهم می کند
  • TensorFlow یک چارچوب طولانی مدت ML با اکوسیستم بزرگ است
  • Reactant.jl چارچوبی برای بهینه سازی و اجرای کد جولیا از طریق OpenXLA، StableHLO و MLIR است.
  • چارچوب GoMLX ML برای زبان Go
    • gopjrt raw XlaBuilder+PJRT wrapper for Go: روی CPU، GPU و TPU تست شده است.

پلاگین های PJRT

  • libTPU به مدل‌ها اجازه می‌دهد روی TPUهای ابری Google اجرا شوند

کامپایل لبه

  • Google AI Edge از StableHLO به عنوان یک قالب ورودی برای استقرار در دستگاه های تلفن همراه با استفاده از LiteRT استفاده می کند
    • AI Edge Torch مدل‌های PyTorch را برای استقرار تلفن همراه از طریق StableHLO صادر می‌کند
  • IREE از StableHLO به عنوان یک قالب ورودی برای استقرار در طیف وسیعی از دستگاه ها و شتاب دهنده ها استفاده می کند
  • StableHLO به CoreML مدل های StableHLO را به CoreML اپل برای استقرار در دستگاه های اپل تبدیل می کند.

ابزار و تجسم

  • Model Explorer تجسم گراف ارثی را با پشتیبانی از مدل های StableHLO ارائه می دهد