機械学習モデルを自動高速化するコンパイラの開発
Description
【概要】
現在様々なアプリケーションで機械学習が利用されています。
これらのアプリケーションを高速化または省電力化するために、様々な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
Requirements
・コンパイラの設計と最適化手法に関する深い知識や経験
・C/C++及びPythonを用いたプログラミング知識や経験
・コンピュータ・サイエンスに対する深い知識
・当該分野に関する興味・関心
Preferred Experiences
・MLIR, LLVM, TVMなどのOSSのコンパイラに精通していること
・PyTorch, TensorFlow, ONNXなどの深層学習フレームワークに精通していること
・CPU, GPU, FPGA, AIアクセラレータなどのハードウェアアーキテクチャに関する知識
・高性能プログラミングの経験
・量子化や枝刈りなどの、モデル圧縮技術に関する知識や研究開発経験
・OSS(Open Sourcre Software)の開発や貢献の実績
We are looking for
・当該分野に関する興味関心があり、その分野での課題解決に貢献したい人
・社会やプロジェクトの現状から課題を発見し、定義出来る人
・発見した課題に対する解決方法を考案し、提案出来る人
・リーダーシップやコミュニケーション能力を活用して、チームワークに貢献できる人
・高度なプログラミングスキルを活用して、課題を解決するプロダクトやソリューションを実現できる人
・当該分野における最新のOSS実装や論文、書籍などを常にキャッチアップし、プロジェクトに還元できる人
Working Conditions
Salary | 理論年収540万円~ ※スキル・ご経験により応相談 ※上記金額は、ベース年収に特別賞与(40万円/年以上)を加えた数字となります。 諸手当: ・通勤手当:上限6万円/月 ・資格手当 ・子供手当:一人につき5千円/月 ・特別賞与:年2回 (※)40万円/年以上(金額は業績によって変動) ・社外常駐手当:4万円/月 ・社員旅行の旅費・滞在費 |
---|---|
Location | 株式会社フィックスターズ本社 東京都港区芝浦3-1-1 msb Tamachi 田町ステーションタワーN 28階 【就業場所の変更の範囲】 会社の定める場所(配置転換、出向、転籍の可能性あり) |
Job Type | 正社員 |
Work hours | 裁量労働制適用 所定労働時間:8時間 (推奨:始業時間10:00、終業時間19:00) 休憩時間:1時間 ※専門業務型裁量労働制により9時間働いたものとみなします 完全週休2日制(土・日・祝日) 、夏季休暇、年末年始休暇 リモートワークについて: 一部可 ※制度としては週2日までが上限となります。 |
Probation period | あり(3ヶ月) |
Benefits | 各種制度: ・社員持株会制度:持株会奨励金として拠出金額の100%を補助 ・社外活動費補助:社外セミナー、勉強会、プログラミングコンテスト、国際学会等の参加費用補助 ・語学研修補助:オンライン英会話または日本語レッスン費用の半分を補助 ・ビジネススキルアップサポート:MBAやPhDの取得希望者に学費等のサポート ・資格取得補助:業務に関連のある資格に対し、合格奨励金の支給や受験料を補助 ・産業医によるメンタルヘルスサポート ・ドリンク補助 ・書籍購入補助 ・キーボードなどの備品購入補助 その他: ・残業時間は月平均20時間 ・「エンジニアが主役」がモットーの会社 ・Google Workspaceの各種アプリの他、コミュニケーションツールとしてSlack、プロジェクト管理としてGitLabを全社で利用 ・勉強会、社内大学等、エンジニアの活動を奨励する成長支援 ・部活動(スキー部、ゴルフ部、テニス部、フットサル部、ボードゲーム部)、おやつタイム、社内新聞などの社内交流文化 |
Company information
Name | 株式会社フィックスターズ |
---|---|
Established | 2002年8月 |
Address | 東京都港区芝浦3-1-1 msb Tamachi 田町ステーションタワーN 28階 |
Capital | 5億5446万円 |
Employees | 320名(2024年9月末現在) |