Junior GPU Software Engineer — Equity Eligible

OXMIQ Labs, Inc.; Campbell (CA); On-site; USD 90,000 - 140,000; Full time

Benefits offered by this job

Equity participation

Medical, dental, and vision coverage

Competitive compensation package

Job summary

OXMIQ Labs, Inc. is seeking an early-career software engineer to contribute to the GPU software stack, including OxPython and kernel/runtime layers.

You will work with senior engineers to develop, test and optimize kernels, runtimes, and enablement layers that deliver high-performance AI workloads on Oxmiq GPU IP. You will learn SIMT concepts, profiling, and tooling while enabling new models and operators across pre-silicon and emulation platforms.

Qualifications

• 1-4 years of experience in GPU, systems or performance-focused software.
• Strong fundamentals in C/C++ and systems programming.
• Hands-on experience with GPU programming, including CUDA.

Responsibilities

• Develop, test and optimize GPU software within OxPython, across kernel, runtime and framework layers.
• Write and tune GPU kernels (CUDA, Triton-based) for AI workloads.
• Profile and benchmark workloads to identify bottlenecks and improve hardware utilization.
• Learn SIMT concepts and memory hierarchy for performance improvements.
• Enable new models and workloads on Oxmiq GPU IP, including pre-silicon bring-up.
• Collaborate with compiler engineers on kernel lowering and mapping via MLIR/LLVM.
• Support validation, debugging and issue triage for software/hardware.
• Perform root cause analysis for functional and performance issues in kernels.
• Contribute to tooling, automation and documentation to streamline development.

Skills

C/C++ programming

GPU programming (CUDA)

SIMT concepts

Profiling & optimization

Debugging & troubleshooting

System-level thinking

PyTorch/JAX/ONNX familiarity

Education

BS/MS/PhD in Computer Science/Engineering

Tools

Nsight

LLVM

MLIR

Triton

CUDA toolkit

Job description

OXMIQ Labs, Inc. is seeking an early-career software engineer to contribute to the GPU software stack, including OxPython and kernel/runtime layers.

You will work with senior engineers to develop, test and optimize kernels, runtimes, and enablement layers that deliver high-performance AI workloads on Oxmiq GPU IP. You will learn SIMT concepts, profiling, and tooling while enabling new models and operators across pre-silicon and emulation platforms.