Oxmiq Labs•10 days ago
Junior GPU Software Engineer
Recruiter Fit Breakdown & Candid Summary
1
Oxmiq Labs is hiring a Junior GPU Software Engineer to work on their OxPython software stack in Hyderabad, India.
2
The role requires 1-4 years of experience, including internships, research, or graduate work, with strong fundamentals in C/C++ and CUDA programming.
3
Candidates will collaborate closely with senior engineers and architects on GPU kernels, runtimes, and performance tuning for AI workloads.
4
This position is best suited for early-career systems or performance engineers passionate about hardware-software co-design and parallel computing.
Role Responsibilities
- 1Develop, test, and optimize GPU software within the OxPython stack across kernel, runtime, and framework integration layers.
- 2Write and tune GPU kernels using CUDA, Triton, or custom ops for AI workloads like GEMM, attention, and convolutions.
- 3Profile and benchmark workloads to identify performance bottlenecks and improve hardware utilization.
- 4Enable new models and customer workloads on Oxmiq GPU IP, supporting bring-up on pre-silicon and emulation platforms.
- 5Participate in design and code reviews while collaborating with compiler engineers on MLIR/LLVM lowering maps.
Skills Matrix
Must-Have Skills
C++(C/C++)
CUDA
GPU Programming
Systems Programming
Python
Nice-to-Have Skills
Triton
LLVM
MLIR
PyTorch
NVIDIA Nsight(Nsight)
Free Instant Match Check
Will an ATS filter reject your resume for this role?
Check your keyword match score, missing must-have skills, and formatting flags — before you apply.
Checking saved resume…
No signups. Free check in 3 seconds.
Ready to submit your application?
Apply directly on Oxmiq Labs's official job portal.