AI Framework and Kernel programming Engineer/ML Engineer- Level 2/3

Development and refinement of models in standard AI frameworks like PyTorch, Tensor flow. Work closely with AI Compiler and hardware accelerator teams and add support for compiler features covering optimization algorithms, code generation, etc. to fully utilize the hardware features for maximum efficiency. Be well acquainted with the latest trends in ML models and compiler technologies to build innovative solutions in our products.

1. Develop and maintain highly efficient low-level parallel compute kernels for NN operators to support generative AI model architectures (LLM, CV, etc.)
2. Integrate optimized kernels into machine learning frameworks for training and inference workloads
3. Optimization of NN (Neural networks) work-loads in standard ML frameworks like PyTorch 
4. Conduct performance profiling, identify hotspots and resolve performance issues
5. Develop and validate test cases for stability and performance measurement of kernels

Overview

• Industry - HR Staffing & Recruitment Firms / Employment Agency / Background - Reference Check Firms , STRATEGY / MANAGEMENT CONSULTING FIRMS, Management Consulting
• Functional Area - IT Software Programming / Analysis / Quality / Testing / Training, IT Robotics & Automation / IoT / Machine Learning
• Job Role - Machine Learning Engineer, Machine Learning Engineer (ML, Azure ML, Azure Monitor), Other IT Robotics & Automation / IoT / Machine Learning, Other Top Management IT Robotics & Automation / IoT / Machine Learning
• Employment type - Full Time - Permanent
• Work Mode - In Office

Qualifications

• Any Graduate - Any Specialization
• Any Post Graduate - Any Specialization
• Any Doctorate - Any Specialization

Job Related Keywords