Cerebras
Cerebras
Cerebras mengembangkan perangkat keras AI berskala wafer dan perangkat lunak terintegrasi untuk pelatihan serta inferensi model berperforma tinggi. Wafer-Scale Engine miliknya menggerakkan sistem CS-3 dan Cerebras Inference Cloud, yang mendukung alur kerja kompatibel dengan OpenAI API serta penerapan di lingkungan on-premise, cloud, dan edge. Platform perusahaan ini dirancang untuk berbagai aplikasi, termasuk penyajian model, fine-tuning, pra-pelatihan, agen AI waktu nyata, genomika, penemuan obat, keamanan siber, dan pencarian perusahaan. Cerebras memadukan chip, sistem, dan perangkat lunak khusus untuk organisasi yang membangun serta mengoperasikan beban kerja kecerdasan buatan yang kompleks, dengan tim beranggotakan 501–1.000 karyawan.

Insinyur Pembelajaran Mesin, Bring-Up Model

Membawa model transformer ke perangkat keras akselerator AI Cerebras dan mengoptimalkan kinerja inferensinya. Mengembangkan jalur compiler MLIR sambil meningkatkan ketepatan model, latensi, throughput, dan efisiensi memori.

Deskripsi

  • Membawa model baru ke produksi dengan menganalisis arsitektur, mengonversi bobot, mengimplementasikan jalur eksekusi, dan memvalidasi hasil terhadap implementasi referensi.
  • Menurunkan model ke perangkat keras Cerebras menggunakan dialek MLIR, transformasi graf, pass lowering, dan pemetaan khusus perangkat keras.
  • Mendukung attention, perkalian matriks, normalisasi, embedding posisi, dan operator lainnya melalui pengembangan compiler dan kernel.
  • Mengoptimalkan fusi operator, tata letak tensor, tiling, alokasi memori, perpindahan data, dan eksekusi paralel.
  • Meningkatkan kinerja prefill dan decode dengan menyetel pengelolaan KV-cache, batching, dan kuantisasi untuk menghasilkan latensi, throughput, serta penggunaan memori yang lebih baik.
  • Menganalisis perbedaan numerik dan menilai pengaruh perubahan presisi serta optimisasi compiler terhadap akurasi.
  • Menggunakan profiler, jejak eksekusi, dan penghitung perangkat keras untuk menemukan hambatan komputasi, memori, komunikasi, dan runtime.
  • Berkolaborasi dengan tim perangkat keras, compiler, kernel, dan runtime untuk menghadirkan dukungan model yang andal serta tolok ukur kinerja yang dapat direproduksi.

Persyaratan

  • Mahir dalam pemrograman C++ dan Python.
  • Berpengalaman langsung membawa dan men-debug model pembelajaran mesin di PyTorch atau framework yang sebanding.
  • Memiliki pengalaman praktis dengan MLIR, termasuk dialek, pola rewrite, pass transformasi, dan pipeline lowering.
  • Memahami dasar-dasar compiler, termasuk representasi perantara, analisis aliran data, dan pembuatan kode.
  • Memahami arsitektur transformer, mekanisme attention, operasi tensor, dan presisi numerik.
  • Berpengalaman membuat profil dan mengoptimalkan workload pada GPU atau akselerator AI lainnya.
  • Mampu mendiagnosis masalah ketepatan dan kinerja di seluruh kode model, keluaran compiler, kernel, dan eksekusi runtime.
  • Berpengalaman dengan inferensi large language model, termasuk GQA, sliding-window attention, mixture-of-experts, KV caching, dan speculative decoding.
  • Berpengalaman dengan FP16, BF16, FP8, atau kuantisasi bit rendah serta kompromi akurasi dan kinerjanya.
  • Memiliki latar belakang dalam pengembangan kernel akselerator atau backend compiler khusus perangkat keras.
  • Memahami eksekusi terdistribusi, paralelisme model, dan hierarki memori akselerator.
  • Memiliki kontribusi pada MLIR, LLVM, framework inferensi, atau proyek open source terkait.

Tunjangan

  • Pekerjaan yang stabil dalam lingkungan dengan ritme startup.
  • Kesempatan untuk menerbitkan dan membuka sumber penelitian AI tingkat lanjut.
  • Bekerja dengan salah satu superkomputer AI tercepat di dunia.
  • Budaya yang lugas dan tidak korporat, serta menghormati keyakinan pribadi.
  • Pembelajaran berkelanjutan, pengembangan profesional, dan dukungan.
  • Tempat kerja yang adil dan beragam.

Lowongan Terkait

AT&T

Perwakilan Penjualan Bisnis AT&T, Akun Wilayah — Memphis

AT&T

Kembangkan basis pelanggan bisnis kecil dan pasar menengah AT&T di Memphis dengan menjual broadband serat optik dan solusi komunikasi lainnya. Peran ini mencakup pencarian prospek, pengembangan akun, dan penjualan konsultatif untuk meningkatkan pendapatan wilayah.

Buka