Junior GPU Software Engineer

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 provides complete hardware and software GPU IP, so our customers can build their own AI silicon. Our OxCore architecture combines scalar, tensor and SIMT engines on one platform. It delivers CUDA compatibility alongside a flexible, fully programmable architecture. OxPython, our software stack, runs on top of it and lets CUDA-optimized models run unmodified on Oxmiq's GPU IP, giving day-zero support for new AI models. We have an established software team and a working stack. We're now moving from development to production delivery, hardening the platform and scaling it to meet real customer demands.

You'll join the OXMIQ software team as an early-career engineer working on the GPU software stack. You'll work closely with senior engineers and architects to build, test and optimize the kernels, runtime and enablement layers that bring high performance to our customers' AI workloads.

Key Responsibilities

• Develop, test and optimize GPU software within OxPython, the OXMIQ software stack, across the kernel, runtime and framework integration layers.


• Write and tune GPU kernels (CUDA, Triton-based and custom ops) for AI workloads such as GEMM, attention, convolution and memory-bound operators.


• Profile and benchmark workloads, identify performance bottlenecks and help close the gap to peak hardware utilization.


• Learn and apply SIMT and parallel compute concepts, including thread/warp scheduling, divergence, occupancy and efficient use of the memory hierarchy.


• Help enable new models, operators and customer workloads on Oxmiq GPU IP, including bring-up on pre-silicon and emulation platforms.


• Work with compiler engineers on how kernels are lowered and mapped onto Oxmiq hardware through MLIR/LLVM.


• Support software/hardware validation through test development, debugging and issue triage.


• Carry out root cause analysis for functional and performance issues in kernels and runtime code.


• Contribute to tooling, automation and documentation that make the stack easier to develop, test and ship.


• Take part in design and code reviews, and learn from senior engineers across the kernels, runtime and frameworks teams.

Required Qualifications

• 1-4 years of experience in GPU, systems or performance-focused software, including internships, research or graduate project work.


• Strong fundamentals in C/C++ and systems programming.


• Hands-on experience with GPU programming, including CUDA, from coursework, research, internships or projects.


• Understanding of SIMT and parallel compute fundamentals: threads/warps, divergence, synchronization, occupancy and memory hierarchies.


• Interest in performance work, plus basic experience with profiling and optimizing code.


• Familiarity with computer architecture and operating systems concepts.


• Exposure to ML frameworks (PyTorch, JAX, ONNX) and how deep learning workloads run on GPUs.


• Strong problem-solving and debugging skills, eagerness to learn, and a bias toward shipping. You prototype, measure and iterate quickly.

Preferred Qualifications

• Experience writing or optimizing GPU kernels such as GEMM, convolution, reductions or attention.


• Exposure to Triton or similar high-level GPU kernel frameworks.


• Familiarity with compiler infrastructure such as LLVM or MLIR.


• Exposure to GPU profiling tools such as Nsight or similar.


• Experience with Claude Code or equivalent AI-assisted development workflows.


• Coursework, research or projects in parallel computing, GPU architecture, compilers or high-performance computing.


• Background in AI workloads and software stacks: LLM inference and training, attention and KV-cache optimization, distributed and multi-GPU execution, or serving frameworks such as vLLM and SGLang.


• Contributions to open-source GPU, compiler, runtime or ML-framework projects such as LLVM, MLIR, Triton or PyTorch.

Education

BS/MS/PhD in Computer Science, Computer Engineering, Electrical Engineering or a related field.

Why OXMIQ

You'll join a fast-growing company with an established team and a proven stack, as we scale it into a production platform in customers' hands. You'll work directly with the people designing the hardware it runs on. This role suits an early-career engineer who wants to build deep GPU software skills, learn from experienced engineers and architects, and work across an entire SIMT compute platform rather than just one layer of it.

OXMIQ offers a competitive compensation package, including base salary, equity participation, comprehensive medical, dental, and vision coverage, and the opportunity to contribute to foundational silicon and software technology.

OXMIQ is an equal opportunity employer. We evaluate qualified applicants without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, veteran status, or any other legally protected characteristic.