This job offer is not available in your country.

AI Software Stack Developer

OXMIQ Labs IncGreater Hyderabad Area, India

11 hours ago

Job description

We're seeking software engineers with hands-on experience in PyTorch C++ source code development, custom backend integration, low-level kernel implementation, and deep understanding of PyTorch scaling library internals. The role requires expertise in C++, Python bindings, CUDA programming, and CMake build systems. Ideal candidates will have contributed to or extensively worked with the core PyTorch codebase and distributed frameworks for large-scale model deployment and orchestration.

Skillsets needed

C++, Python bindings, CMake
Triton lang, CUDA, PTX, or equivalent on any HW
Leverage open-source tools

Experience Needed

Hands-on experience in PyTorch C++ source code and / or custom backend integration

Low-level kernel implementation

Deep understanding of PyTorch scaling library internals

Contributed to or extensively worked with the core PyTorch codebase and distributed frameworks for large-scale model deployment and orchestration

MS with 3+ years or BS with 5+ years or excellent contributions to open-source projects

Create a job alert for this search

Ai Developer • Greater Hyderabad Area, India