Software Engineer, AI Training and Infrastructure

About the role

We are looking for a Software Engineer to work at the forefront of developing and optimizing the software infrastructure and tools necessary for training cutting-edge AI models. You will focus on building robust, scalable, and efficient training pipelines and frameworks that support the entire machine learning lifecycle, from data preparation to model deployment.

What you'll do

• Develop and maintain robust, scalable, and distributed training pipelines (data preprocessing, training orchestration
• Optimize training processes for performance and resource utilization, ensuring scalability and reliability
• Collaborate with researchers and machine learning engineers to integrate state-of-the-art algorithms and techniques into training pipelines
• Monitor and analyze training, identifying bottlenecks and proposing solutions to improve efficiency and performance
• Ensure the robustness and reliability of the training infrastructure, including automated testing and continuous integration

What you need

• 3–6 yrs of experience
• Data structures
• Machine learning
• Test automation
• B.Tech or B.E. or B.Sc or BCA

Good to have

• System design
• Python
• C++
• AWS
• Deep learning
• GCP