業務内容
【概要】
機械学習モデルを自動高速化するコンパイラの開発
現在様々なアプリケーションで機械学習が利用されています。
これらのアプリケーションを高速化または省電力化するために、様々な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