ML Engineer — Distributed Training Systems ML Platform

Highlanders Inc.; San Francisco (CA); On-site; USD 150,000 - 230,000; Full time

Benefits offered by this job

社会保険完備

英会話手当

引越し手当

Job summary

Highlanders Inc.は、分散学習システムの分野で最先端の基盤モデルを開発しています。本ポジションは複数のGPUを用いた学習環境の性能と安定性を向上させ、データ並列・モデル並列の両方のアプローチを実装します。

現場運用と研究の両方を担い、通信・メモリの最適化、障害の切り分け・改善を推進できるエンジニアを求めています。日本と米国拠点の双方での協働が想定されます。

Qualifications

• 分散学習の実装・最適化経験が必須。データ並列・モデル並列の知識があること。
• NCCL・CUDA等の低レイヤー最適化の経験が望ましい。

Responsibilities

• 分散学習手法の実装と最適化を担当。
• 通信・メモリの効率化を推進。
• 学習障害の切り分けと改善を行う。

Skills

NCCL

CUDA

Distributed training

Performance optimization

Debugging

Tools

PyTorch

Job description

ML Engineer — Distributed Training Systems（分散学習システム）

Description

Highlanders は、ヒューマノイドロボットと、フィジカルAI（現実の世界で体を動かすAI）の基盤モデル、そしてその学習データを生み出すロボットの群れを、一つの流れとして開発しています。実機が現場で働くほどデータが集まり、モデルが賢くなり、ロボットがさらに働けるようになる。このデータの循環を、日本の工場の現場とサンフランシスコの両方から回しています。

多数の GPU を使った分散学習の性能と安定性を担っていただきます。

お任せしたいこと

• 分散学習の手法（データ並列・モデル並列など）の実装と最適化
• 通信・メモリの効率化
• 学習の障害の切り分けと改善

Requirements

Highlanders は、ヒューマノイドロボットと、フィジカルAI（現実の世界で体を動かすAI）の基盤モデル、そしてその学習データを生み出すロボットの群れを、一つの流れとして開発しています。実機が現場で働くほどデータが集まり、モデルが賢くなり、ロボットがさらに働けるようになる。このデータの循環を、日本の工場の現場とサンフランシスコの両方から回しています。

多数の GPU を使った分散学習の性能と安定性を担っていただきます。

お任せしたいこと

• 分散学習の手法（データ並列・モデル並列など）の実装と最適化
• 通信・メモリの効率化
• 学習の障害の切り分けと改善

Preferred Experiences

• NCCL、CUDA などの低レイヤーの最適化経験

We are looking for

• 性能の数字にこだわり、ボトルネックを突き止められる方

Salary

ご経験・スキルを考慮のうえ、面談にてご提示します。

Location

San Francisco, CA（米国）または所沢開発拠点（埼玉県所沢市）
※どちらの拠点での勤務も可能です。ご希望をお聞かせください。

Job Type

正社員（業務委託・副業での参画もご相談可）

Work hours

勤務時間：9:00〜18:00
休日・休暇：完全週休二日制

Probation period

6か月

Benefits

• 社会保険完備（健康保険、厚生年金保険、雇用保険、労災保険）
• 英会話手当
• 引越し手当