3 months ago
Bengaluru, India +2 moreMid Level / Senior
H1B Sponsor
Base Salary
$100k - $300k/yr
Responsibilities
- Develop and maintain robust, scalable, and distributed training pipelines for AI models.
- Optimize training processes for performance and resource utilization.
- Collaborate with researchers and machine learning engineers to integrate algorithms into training pipelines.
- Monitor and analyze training to identify bottlenecks and propose efficiency improvements.
- Ensure the robustness and reliability of the training infrastructure through automated testing.
Requirements
- BS, MS or higher degree in Computer Science, Robotics, Engineering or a related field, or equivalent practical experience.
- Minimum of 3 years of industry experience.
- Proficiency in Python, C++, or similar and at least one deep learning library such as PyTorch, TensorFlow, or JAX.
- Strong background in distributed computing and parallel processing techniques.
- Deep understanding of state-of-the-art machine learning techniques and models.
- Experience with cloud-based training environments like AWS, Google Cloud, or Azure.
- Experience in developing and maintaining software tooling and infrastructure for machine learning.
- Deep understanding of software engineering principles, including algorithms and system design.
- Experience with continuous integration and automated testing frameworks.