Job Title: ML Benchmarking Engineer
Location: Bengaluru, Karnataka
Job Type: Full-Time
Experience Required: 3–7 Years
Education
Bachelor's or Master's degree in Computer Science, Electronics, Electrical Engineering, Embedded Systems, Artificial Intelligence, Machine Learning, or a related field.
About the Role
To support the evaluation and optimization of machine learning models on embedded and hardware-based platforms. The role focuses on performance benchmarking, workflow automation, and validation of AI workloads across specialized hardware environments.
The successful candidate will work closely with software, hardware, and machine learning teams to measure, analyze, and improve model performance and deployment efficiency.
Key Responsibilities
- Evaluate and benchmark machine learning models on hardware platforms.
- Measure and analyze performance metrics including latency, throughput, resource utilization, and accuracy.
- Execute validation activities for AI/ML workloads on embedded and hardware-based environments.
- Develop and maintain automated testing and benchmarking frameworks.
- Support benchmarking and validation workflows through automation and process improvements.
- Investigate performance issues and conduct root cause analysis.
- Maintain documentation related to testing methodologies, results, and technical findings.
- Collaborate with cross-functional engineering teams to support model deployment and optimization activities.
Required Skills
- 3+ years of experience in benchmarking ML models on hardware platforms.
- Proficiency in Python and C/C++.
- Hands-on experience with PyTorch, TensorFlow, or similar ML frameworks.
- Experience working with hardware accelerators and custom SoCs.
- Strong debugging and problem-solving abilities.
- Excellent teamwork and communication skills.
Preferred Experience
- Experience with embedded AI, edge AI, or DSP-based ML deployments.
- Exposure to performance optimization and profiling tools.
- Familiarity with AI inference acceleration technologies.
- Experience working in cross-functional hardware and software development environments.
Pay: ₹2,100,000.00 - ₹2,200,000.00 per year
Benefits:
- Health insurance
- Life insurance
- Provident Fund
Work Location: In person
