Langsung ke konten utama
OpenXLA Project
XLA
  • Pemasangan/Penggunaan
  • Dokumentasi
  • Kode Sumber
StableHLO
  • Dokumentasi
  • Kode Sumber
Shard
  • Dokumentasi
  • Kode Sumber
PJRT
  • Dokumentasi
  • Kode Sumber
XProf
  • Dokumentasi
  • Kode Sumber
Tokamax
  • Dokumentasi
  • Kode Sumber
Komunitas
  • JAX & OpenXLA Summer Dev Lab 2025
  • Peristiwa & Rekaman
  • Rapat Komunitas
  • OpenXLA Discuss
  • Discord
Partner Industri Kontak
/
  • English
  • Deutsch
  • Español – América Latina
  • Français
  • Indonesia
  • Italiano
  • Polski
  • Português – Brasil
  • Tiếng Việt
  • Türkçe
  • Русский
  • עברית
  • العربيّة
  • فارسی
  • हिंदी
  • বাংলা
  • ภาษาไทย
  • 中文 – 简体
  • 中文 – 繁體
  • 日本語
  • 한국어
Masuk
  • Tokamax
OpenXLA Project
  • XLA
    • Lainnya
  • StableHLO
    • Lainnya
  • Shard
    • Lainnya
  • PJRT
    • Lainnya
  • XProf
    • Lainnya
  • Tokamax
    • Lainnya
  • Komunitas
    • Lainnya
  • Partner Industri
  • Kontak
  • Tokamax
  • Pengantar
  • Panduan Developer
    • Hardware dan Operasi yang Didukung
    • Penyesuaian otomatis
    • Tolok Ukur dan Angka
    • Berkontribusi
  • Notebook Demo
    • Demo Tokamax
  • Dokumentasi khusus Op
    • Perhatian Splash
  • Pemasangan/Penggunaan
  • Dokumentasi
  • Kode Sumber
  • Dokumentasi
  • Kode Sumber
  • Dokumentasi
  • Kode Sumber
  • Dokumentasi
  • Kode Sumber
  • Dokumentasi
  • Kode Sumber
  • Dokumentasi
  • Kode Sumber
  • JAX & OpenXLA Summer Dev Lab 2025
  • Peristiwa & Rekaman
  • Rapat Komunitas
  • OpenXLA Discuss
  • Discord
  • OpenXLA
  • Tokamax
Tetap teratur dengan koleksi Simpan dan kategorikan konten berdasarkan preferensi Anda.

Hardware dan Operasi yang Didukung

Saat ini kami mendukung kernel GPU berikut:

  • tokamax.dot_product_attention (FlashAttention).
  • tokamax.gated_linear_unit (Unit linear yang di-gate (SwiGLU, dll)).
  • tokamax.layer_norm (Layer normalization dan Root Mean Squared normalization).

Dan berikut untuk GPU dan TPU:

  • tokamax.ragged_dot (Mixture of Experts).

Kecuali dinyatakan lain, konten di halaman ini dilisensikan berdasarkan Lisensi Creative Commons Attribution 4.0, sedangkan contoh kode dilisensikan berdasarkan Lisensi Apache 2.0. Untuk mengetahui informasi selengkapnya, lihat Kebijakan Situs Google Developers. Java adalah merek dagang terdaftar dari Oracle dan/atau afiliasinya.

Terakhir diperbarui pada 2025-11-21 UTC.

  • Tetap terhubung

    • Twitter
    • YouTube
  • Persyaratan
  • Privasi
  • Manage cookies
  • English
  • Deutsch
  • Español – América Latina
  • Français
  • Indonesia
  • Italiano
  • Polski
  • Português – Brasil
  • Tiếng Việt
  • Türkçe
  • Русский
  • עברית
  • العربيّة
  • فارسی
  • हिंदी
  • বাংলা
  • ภาษาไทย
  • 中文 – 简体
  • 中文 – 繁體
  • 日本語
  • 한국어