Full-Stack AI Compute Architect
Hyderabad, India | On-site
Experience: 12 + Years
About OXMIQ
OXMIQ provides complete hardware and software GPU IP that lets our customers build their own AI silicon. Founded by Raja Koduri, we recently closed a $35M Series A (co-led by Samsung Catalyst Fund and Fundomo, with MediaTek, Intel Capital, and others), bringing total funding to $60M, with Jim Keller joining our board.
Our OxCore architecture combines scalar, tensor, SIMT, and orchestration engines on one platform — delivering CUDA compatibility alongside a fully programmable architecture. On top of it runs OxPython, our software stack that runs existing CUDA and PyTorch code unmodified with day-zero support for recent AI models. We have an established team and a working stack, and we're now shifting from development to production — hardening the platform and scaling it for inference at real customer scale.
You'll join the OXMIQ architecture team, which sets the direction of our software, silicon, and systems.
You'll help refine our software strategy and deliver the highest-performance solutions by optimizing the whole stack — from AI models and frameworks at the top down to kernel and hardware-level code generation.
The Scope of This Role
This role spans the AI software stack from models on down — the full vertical path a workload travels: from a model authored in frameworks such as PyTorch, JAX, or ONNX, through graph capture and the compiler, into the runtime and scheduler, down to individual kernels and the instructions that execute on OxCore.
We're looking for an architect who is fluent across that whole range — someone who can reason about how a transformer or diffusion model is structured and about warp divergence and register allocation, and who can make the choices that connect those layers into one coherent, high-performance platform. You won't be the deepest specialist at every layer, but you should be able to hold the whole range in your head and drive decisi