機械学習モデルを自動高速化するコンパイラの開発
仕事概要
【概要】
現在様々なアプリケーションで機械学習が利用されています。
これらのアプリケーションを高速化または省電力化するために、様々な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
【プロジェクトのやりがい】
・急成長するAI/ML市場において、新たな技術革新にチャレンジできる
・AIコンパイラやAIフレームワークに関する、最先端かつ大規模なコンピュータシステムの開発に携われる
・著名なOSSを利用した開発やコントリビューションが出来る
・次世代のAIアクセラレータの設計や開発に携われる
【開発環境】
開発環境: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実装や論文、書籍などを常にキャッチアップし、プロジェクトに還元できる人
応募概要
給与 | 理論年収540万円~ ※スキル・ご経験により応相談 ※上記金額は、ベース年収に特別賞与(40万円/年以上)を加えた数字となります。 諸手当: ・通勤手当:上限6万円/月 ・資格手当 ・子供手当:一人につき5千円/月 ・特別賞与:年2回 (※)40万円/年以上(金額は業績によって変動) ・社外常駐手当:4万円/月 ・社員旅行の旅費・滞在費 |
---|---|
勤務地 | 株式会社フィックスターズ本社 東京都港区芝浦3-1-1 msb Tamachi 田町ステーションタワーN 28階 【就業場所の変更の範囲】 会社の定める場所(配置転換、出向、転籍の可能性あり) |
雇用形態 | 正社員 |
勤務体系 | 裁量労働制適用 所定労働時間:8時間 (推奨:始業時間10:00、終業時間19:00) 休憩時間:1時間 ※専門業務型裁量労働制により9時間働いたものとみなします 完全週休2日制(土・日・祝日) 、夏季休暇、年末年始休暇 リモートワークについて: 一部可 ※制度としては週2日までが上限となります。 |
試用期間 | あり(3ヶ月) |
福利厚生 | 各種制度: ・社員持株会制度:持株会奨励金として拠出金額の100%を補助 ・社外活動費補助:社外セミナー、勉強会、プログラミングコンテスト、国際学会等の参加費用補助 ・語学研修補助:オンライン英会話または日本語レッスン費用の半分を補助 ・ビジネススキルアップサポート:MBAやPhDの取得希望者に学費等のサポート ・資格取得補助:業務に関連のある資格に対し、合格奨励金の支給や受験料を補助 ・産業医によるメンタルヘルスサポート ・ドリンク補助 ・書籍購入補助 ・キーボードなどの備品購入補助 その他: ・残業時間は月平均20時間 ・「エンジニアが主役」がモットーの会社 ・Google Workspaceの各種アプリの他、コミュニケーションツールとしてSlack、プロジェクト管理としてGitLabを全社で利用 ・勉強会、社内大学等、エンジニアの活動を奨励する成長支援 ・部活動(スキー部、ゴルフ部、テニス部、フットサル部、ボードゲーム部)、おやつタイム、社内新聞などの社内交流文化 |
企業情報
企業名 | 株式会社フィックスターズ |
---|---|
設立年月 | 2002年8月 |
本社所在地 | 東京都港区芝浦3-1-1 msb Tamachi 田町ステーションタワーN 28階 |
資本金 | 5億5446万円 |
従業員数 | 320名(2024年9月末現在) |