Cerebrasは、高性能なモデルの学習と推論に向けたウェハースケールAIハードウェアと統合ソフトウェアを開発しています。同社のウェハースケールエンジンは、CS-3システムとCerebras Inference Cloudを支え、OpenAI API互換のワークフローや、オンプレミス、クラウド、エッジ環境への展開に対応しています。このプラットフォームは、モデルサービング、ファインチューニング、事前学習、リアルタイムAIエージェント、ゲノミクス、創薬、サイバーセキュリティ、エンタープライズ検索などの用途向けに設計されています。Cerebrasは、専門チップ、システム、ソフトウェアを組み合わせ、負荷の高い人工知能ワークロードを構築・運用する組織を支援しており、従業員数は501～1,000人です。
機械学習ランタイム・カーネルエンジニア、コア機械学習
コア機械学習の研究成果を本番ソフトウェアへ変換する、高性能ランタイムとカーネルを開発します。 Cerebras のウェーハスケール AI プラットフォームと機械学習システム向けのソフトウェアを構築・最適化します。
仕事内容
- 新しいコア機械学習アルゴリズム向けのランタイムコンポーネントと高性能カーネルを設計・実装する
- 研究プロトタイプを効率的な Cerebras プラットフォーム実装へ変換し、必要に応じてリファレンス実装や GPU との比較も行う
- 機械学習フレームワーク、コンパイラ、ランタイム、通信システム、カーネル全体の性能をプロファイリングし、問題を解決する
- 大規模トレーニングと低レイテンシ推論に向けて、計算、メモリ転送、通信、並行性を最適化する
- 機能、性能、数値精度を検証するベンチマーク、計測機能、自動テストを作成する
- コア機械学習の研究者、コンパイラ・ランタイム・カーネル・推論エンジニアと緊密に連携する
- 繰り返し発生する制約とプラットフォーム改善の機会を特定し、ソフトウェアアーキテクチャやロードマップの意思決定に貢献する
応募要件
- コンピュータサイエンス、コンピュータエンジニアリング、電気工学、または関連分野の学士・修士・博士号、もしくは同等の実務経験
- 高性能システムソフトウェア、機械学習システム、ランタイム、コンパイラ、または計算カーネルの開発経験
- C++ と Python による高いプログラミング能力
- 並列プログラミング、メモリ管理、並行性、データ構造、性能最適化に関する確かな知識
- 複数のシステム層にまたがる複雑なソフトウェアをデバッグ・プロファイリングした実績
- PyTorch や JAX など、最新の機械学習アーキテクチャとフレームワークに関する知識
- 研究者と効果的に協働し、変化するアルゴリズム上の要件を信頼性の高いソフトウェアへ変換する能力
- 歓迎：CUDA、Triton、低レベルアセンブリ、アクセラレータプログラミング、または C 系のドメイン固有言語の経験
- 歓迎：コンパイラ内部、分散ランタイム、カスタムハードウェアインターフェース、または高性能コンピューティングシステムの経験
- 歓迎：機械学習の基礎と機械学習システムへの理解
- 歓迎：アテンション、KV キャッシュ管理、並列生成、分散実行などを含む、大規模言語モデルのトレーニングまたは推論に関する知識
- 歓迎：産業界または学術研究環境でのソフトウェア開発経験
- 歓迎：大規模なオープンソースシステム、機械学習フレームワーク、コンパイラ、またはカーネルライブラリへの貢献経験
福利厚生
- 従来の GPU の制約を超える AI プラットフォームの構築に携われる
- 先進的な AI 研究を発表し、オープンソース化する機会
- 世界最速クラスの AI スーパーコンピュータの一つに携われる
- スタートアップの活気と雇用の安定性を両立
- 個人の信条を尊重する、率直で非官僚的な文化
- 平等で多様性のある職場環境
- 継続的な学習、成長、サポートの機会