)

JOB ID:110079

非公開求人
非公開求人
給与 540万円 〜
業種 IT・通信・インターネット
勤務地 東京都
業務内容 【概要】
機械学習モデルを自動高速化するコンパイラの開発

現在様々なアプリケーションで機械学習が利用されています。
これらのアプリケーションを高速化または省電力化するために、様々なAIアクセラレータが開発されています。
フィックスターズでは、AIアクセラレータを開発するお客様向けに、AIコンパイラフレームワークの開発を行っています。
AIコンパイラは、機械学習モデルの学習や推論のワークロードを、対象のAIアクセラレータ上で高速処理するための変換とコード生成を自動で適用します。

フィックスターズでは、このAIコンパイラフレームワークを設計及び開発するエンジニアを募集しています。
この役割には、対象のAIアクセラレータを対象にした量子化、計算グラフの最適化、ループやメモリ配置の最適化の手法の設計や、コンパイラへの実装が含まれます。
また、AIアクセラレータやAIコンパイラの最新研究やOSSの動向の調査も行い、それらを用いた技術選定にも携わります。
加えて、ハードウェア設計者とも連携し、ハードウェアとソフトウェアの協調して設計することで、次世代のAIアクセラレータの設計にも貢献していただきます。

大規模かつ最先端のAIコンパイラフレームワークや、次世代のAIアクセラレータの開発に携われるプロジェクトにぜひ参加しませんか?

【具体的な職務内容】
・AIコンパイラフレームワークの設計や開発
・対象のAIアクセラレータ向けの高速化手法の設計
・量子化や枝刈りなどのAIモデル圧縮技法の研究開発や実装
・AIコンパイラフレームワークを利用した機械学習モデルの高速化
・AIアクセラレータのハードウェアアーキテクチャに対する改善手法の提案

【案件例】
https://www.fixstars.com/ja/cases/r-car

【開発環境】
開発環境:C/C++, Python
コンパイラフレームワーク: MLIR, LLVM, TVM
AIフレームワーク: PyTorch, TensorFlow, ONNX
その他開発環境:Linux
開発支援ツール:Git, GitLab, GitHub
開発手法:チケット駆動開発, スクラム開発
開発内容タイプ:B2B
応募資格 必須スキル
・コンパイラの設計と最適化手法に関する深い知識や経験
・C/C++及びPythonを用いたプログラミング知識や経験
・コンピュータ・サイエンスに対する深い知識
・当該分野に関する興味・関心

歓迎スキル
・MLIR, LLVM, TVMなどのOSSのコンパイラに精通していること
・PyTorch, TensorFlow, ONNXなどの深層学習フレームワークに精通していること
・CPU, GPU, FPGA, AIアクセラレータなどのハードウェアアーキテクチャに関する知識
・高性能プログラミングの経験
・量子化や枝刈りなどの、モデル圧縮技術に関する知識や研究開発経験
・OSS(Open Sourcre Software)の開発や貢献の実績

求める人物像
・当該分野に関する興味関心があり、その分野での課題解決に貢献したい人
・社会やプロジェクトの現状から課題を発見し、定義出来る人
・発見した課題に対する解決方法を考案し、提案出来る人
・リーダーシップやコミュニケーション能力を活用して、チームワークに貢献できる人
・高度なプログラミングスキルを活用して、課題を解決するプロダクトやソリューションを実現できる人
・当該分野における最新のOSS実装や論文、書籍などを常にキャッチアップし、プロジェクトに還元できる人
福利厚生 / 待遇 各種制度:
・社員持株会制度:持株会奨励金として拠出金額の100%を補助
・社外活動費補助:社外セミナー、勉強会、プログラミングコンテスト、国際学会等の参加費用補助
・語学研修補助:オンライン英会話または日本語レッスン費用の半分を補助
・ビジネススキルアップサポート:MBAやPhDの取得希望者に学費等のサポート
・資格取得補助:業務に関連のある資格に対し、合格奨励金の支給や受験料を補助
・産業医によるメンタルヘルスサポート
・ドリンク補助
・書籍購入補助
・キーボードなどの備品購入補助

その他:
・残業時間は月平均20時間
・「エンジニアが主役」がモットーの会社
・Google Workspaceの各種アプリの他、コミュニケーションツールとしてSlack、プロジェクト管理としてGitLabを全社で利用
・勉強会、社内大学等、エンジニアの活動を奨励する成長支援
・部活動(スキー部、ゴルフ部、テニス部、フットサル部、ボードゲーム部)、おやつタイム、社内新聞などの社内交流文化

リモートワークについて:
一部可
※制度としては週2日までが上限となります。
勤務時間 裁量労働制適用
所定労働時間:8時間 (推奨:始業時間10:00、終業時間19:00)
休憩時間:1時間
※専門業務型裁量労働制により9時間働いたものとみなします
休日休暇 完全週休2日制(土・日・祝日) 、夏季休暇、年末年始休暇