Views:209
Applicants:12
Posted on 17 Mar, 2025
Sorry, this vacancy has been closed and is no longer accepting applications.
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
Job Description | Role and Responsibilities
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.
- Develop and maintain highly efficient low-level parallel compute kernels for NN operators to support generative AI model architectures (LLM, CV, etc.)
- Integrate optimized kernels into machine learning frameworks for training and inference workloads
- Optimization of NN (Neural networks) work-loads in standard ML frameworks like PyTorch
- Conduct performance profiling, identify hotspots and resolve performance issues
- Develop and validate test cases for stability and performance measurement of kernels
Candidate Profile | Who Can Apply
- Good experience in parallel compute programming languages like CUDA, OpenCL, etc.
- Experience or knowledge about BLAS operators and GEMM optimization.
- Knowledge of Computer and GPU architecture.
- Proven experience in kernel development and optimizations
- Familiar with standard ML frameworks (PyTorch, TF, ONNX)
- Knowledge on SOTA LLM models and model architectures will be huge plus
- Knowledge with deep neural network machine learning technologies and modern machine learning Programming frameworks: Experience with performance profiling and optimization tools
- Proficient in C/C++ and Python
- Familiarity with ARM SW ecosystem and toolchains will be a plus
Education
- Any Graduate - Any Specialization
- Any Post Graduate - Any Specialization
- Any Doctorate - Any Specialization
Job Related Keywords
About Employer
Candidate Can Connect
