Junior System Modeling Engineer: GPU AI Simulators

Remote Jobs; United States; Remote; USD 90,000 - 150,000; Full time

Benefits offered by this job

Mentorship

Equity

Competitive compensation

Job summary

Unconventional AI is seeking a Junior Member of Technical Staff, System Modeling, to work with senior engineers on multi‑disciplinary simulation frameworks for unconventional physics‑based AI compute. You will help build GPU‑accelerated simulators and integrate physics‑based models into PyTorch, enabling rapid iteration and cross‑layer optimization.

You will contribute to end‑to‑end simulation environments, ensure reproducibility with robust experiment tracking, and collaborate with Algorithms

Qualifications

• Strong foundation in computer architecture and OS concepts.
• Proficiency in C++ and Python for performance‑critical code.
• Familiarity with deep learning frameworks and basic model concepts.
• Advanced degree or equivalent research experience is a plus.

Responsibilities

• Contribute to GPU‑accelerated simulators for ML on analog/unconventional hardware.
• Integrate physics‑based models into the PyTorch simulation environment.
• Maintain and extend the unified end‑to‑end simulation framework.
• Help implement robust experiment tracking for reproducibility and auditing.
• Collaborate with Algorithms and Hardware teams to align modeling needs.

Skills

C++

Python

Computer Architecture

Operating Systems

Education

BS or higher in CS/EE

MS/PhD preferred

Tools

PyTorch

CUDA

LLVM

MLIR

Job description

Unconventional AI is seeking a Junior Member of Technical Staff, System Modeling, to work with senior engineers on multi‑disciplinary simulation frameworks for unconventional physics‑based AI compute. You will help build GPU‑accelerated simulators and integrate physics‑based models into PyTorch, enabling rapid iteration and cross‑layer optimization.

You will contribute to end‑to‑end simulation environments, ensure reproducibility with robust experiment tracking, and collaborate with Algorithms