仕事内容
PFNの計算基盤の心臓部であるMN-Coreのソフトウェアスタックの研究開発を行うエンジニアを募集します。独自のアクセラレータMN-CoreおよびMN-Core2のコンパイラ開発を通じて、機械学習やシミュレーションの膨大な計算量を支えます。Pythonで定義されたNN (DL) コードをMN-Coreで動作させるためのプロダクト・ツール開発に携わり、グラフ最適化や高速なカーネル実装などを行います。世界最高を目指したアクセラレータ開発に意欲のある方を歓迎します。
Job Description / 職務内容
- PFNの計算基盤の心臓部であるMN-Coreのソフトウェアスタックの研究開発を行う。
- PFNは独自のアクセラレータMN-CoreおよびMN-Core2を開発・活用している。
- MN-Coreの事業加速にはコンパイラなどのソフトウェアスタック開発が非常に重要。
- MN-Coreはプロセッサ制御の大半をソフトウェアで事前にスケジューリングし、高い効率を出すアーキテクチャを採用。
- Pythonで定義されたNN (DL) のコードをMN-Coreで動作させるプロダクト・ツール開発プロジェクトにアサイン。
- 具体的には、グラフ最適化コンパイラ、Numpy-like interface、高速なカーネル実装、コンパイラコア実装に携わる。
- MN-Coreへの高速なデータ供給/前処理、社内ワークロードのMN-Coreへのポーティング/高速化作業も担当。
- 希望や適性に応じてMN-Core自体のアーキテクチャ検討など、高速な計算機を作る・活用する業務に従事可能。
Appeal
- 新しい世界を作り出す仕事に携われる。
- 最先端のDeep Learning向けASICに向けた「実際に使われている」機械学習コンパイラの開発に関わることができる。
- 実際のユーザやハードウェア開発者と距離が近く、必要や興味に応じて様々な仕事ができる。
- 高い技術力を持ったチームメンバーと共に働くことができる。
- HWエンジニアと近い距離で、共に議論しながら開発ができる。
求めるスキル・経験
歓迎
競技プログラミングに関する知識、経験
最適化問題に対する知識、実務経験
機械学習/深層学習の基礎的なアルゴリズムやワークフローの知識、実務経験 機械学習モデル開発経験
複数の深層学習フレームワークの API および実装に関する知識、実務経験
システムプログラミングの経験、実務経験
コンパイラに関する知識、実務経験
一般的なコンパイラに関する知識
コンパイラから出力された機械語(アセンブリ)を直接読み書きするような経験
募集要項
- 雇用形態
- 正社員
- 想定年収
- 応相談
- 勤務地
- 東京都 千代田区
- 出社・リモート
- ハイブリッド / リモート可
- レベル
- メンバー
- 勤務時間
- 裁量労働制 / 実働8時間
- 休日休暇
- 完全週休2日制 / 祝日休み
- 試用期間
- 3ヶ月