Research Engineer, AI Models

Onsite India

pythongopytorch applied research

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

Join a team focused on enhancing AI model capabilities and efficiency in Bangalore. You'll work on cutting-edge techniques to optimize AI inference and collaborate closely with hardware teams.

Details

Bangalore, work from office
5+ years of experience
Strong fundamentals in Python and PyTorch

The work

Research and implement techniques to accelerate AI inference
Partner with hardware teams to optimize algorithmic improvements
Build profiling tools and benchmarking frameworks
Develop fine-tuning workflows for modern AI models