AI Compiler Engineer

Onsite India

pythonpytorchtensorflow software

greenhouse

Match & cover letter

Create a profile to see your match and get a cover letter for this role.

Create profile and match

Job description

You will develop and optimize graph compilers for AI and ML workloads in India. Collaborate with a team to enhance performance and enable efficient model deployment.

Details

India, work from office
3+ years in compiler development
Skills: Python, PyTorch, TensorFlow, C++

The work

Architect and implement optimizations for AI model execution
Collaborate with ML researchers and hardware engineers
Develop compiler optimizations for high-level AI models
Provide leadership and mentorship to engineering teams