Machine Learning Compiler Engineer

2 months ago


Mumbai, Maharashtra, India Wenger & Watson Inc. Full time

Overview :

- Development and support of new AI/ML compiler features/technologies to accelerate deep learning models.

- Work closely with AI 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.

Roles and responsibilities :

- Develop end-to-end ML compiler leveraging standard compiler infrastructures like TVM, MLIR, Torch Dynamo/Inductor taking advantage of both intra-operator parallelism and graph/pipeline/dataflow parallelism while mapping to hardware compute/processing elements of custom AI accelerator

- Implement low level parallel programming model for development/deployment of high performance kernels fully utilizing the hardware capabilities

- Adapt advance techniques/algorithms for placement/scheduling and parallelization of model graphs to improve the performance of ML applications optimizing execution speed and resource utilization

- Code generation leveraging LLVM for the AI accelerator compute elements/cores of targeted ISA with custom instructions

- Debugging and profiling of features of compiler to identify issues and performance hot spots

- Own responsibility throughout the product lifecycle in solving functional/performance issues during execution phase

- Work closely with hardware architecture team for efficient HW/SW Co-design of AI accelerator IP and overall AI server-class SoC

- Keep up-to-date with the industry trend in compiler frameworks in terms of feature advancements, design methods and approaches

Qualifications :

- Should have minimum 12+ years of relevant experience in AI/ML

- Should have deep practical experience of developing end-to-end ML compiler with any AI accelerator architecture (GPU/ASIC/many-core heterogenous)

- Should have good experience with handling complex hierarchies of compute elements and memories of hardware accelerator and their mapping in backend compiler for efficient power/performance

- Should have very good knowledge in any one of compiler frameworks like TVM, MLIR, Torch Dynamo/Inductor or equivalent

- Good knowledge on SOTA Generative AI model architectures like LLM and model optimization techniques will be helpful

- Good knowledge on popular ML framework ecosystems (PyTorch/TensorFlow/ONNX)

- High proficiency in C/C++, Python, domain-specific languages and parallel programming languages like OpenCL/CUDA

(ref:hirist.tech)

  • Mumbai, Maharashtra, India The Nielsen Company Full time

    Job Title: {title}About the Role:We are seeking a highly skilled Machine Learning Engineer to join our team at [Company]. As a Machine Learning Engineer, you will be responsible for designing, developing, and deploying machine learning models to solve complex business problems.Key Responsibilities:Design and develop machine learning models using popular...


  • Mumbai, Maharashtra, India Collinson Full time

    About the RoleCollinson is a global loyalty and benefits company that uses expertise and products to craft customer experiences for some of the world's best-known brands. As a Machine Learning Engineer at Collinson, you will play a critical role in driving the development of cloud-based machine learning pipelines for data-driven products and services.Key...


  • Mumbai, Maharashtra, India Collinson Full time

    About the RoleWe are seeking a highly skilled Machine Learning Engineer to join our team at Collinson. As a key member of our data science and engineering team, you will play a critical role in driving the development of cloud-based machine learning pipelines for data-driven products and services.Key Responsibilities:Design, develop, and deliver machine...


  • Mumbai, Maharashtra, India HUNTINGCUBE RECRUITMENT SOLUTIONS PRIVATE LIMITED Full time

    Job Summary:We are seeking an experienced Machine Learning Engineer to join our team at Huntingcube Recruitment Solutions Private Limited. As a key member of our organization, you will be responsible for designing and developing large-scale machine learning systems that drive business growth.About the Role:This is an exciting opportunity for a skilled...


  • Mumbai, Maharashtra, India Flairchase Full time

    Job Description : Responsibilities : - Design the data pipelines and engineering infrastructure to support machine learning systems at scale.- Take offline models data scientists build and turn them into a real machine learning production system.- Develop and deploy scalable tools and services to handle machine learning training and inference.- Identify and...


  • Mumbai, Maharashtra, India Green s HR Solutions Full time

    Job Title: Software Engineer in Machine LearningWe are seeking a highly skilled Software Engineer in Machine Learning to join our team. The ideal candidate will have expertise in developing and deploying machine learning models using Python.Develop and train machine learning models using Python and popular libraries such as TensorFlow and PyTorchDeploy and...


  • Mumbai, Maharashtra, India Marken Full time

    About the Role:We are seeking an experienced AI/ML engineer who will collaborate closely with cross-functional teams to develop and deploy AI/ML-powered features and tools.Main Responsibilities:Pioneering cutting-edge AI/ML solutions to enhance end-to-end logistics processes.Collaborating seamlessly with cross-functional teams to craft machine learning...


  • Mumbai, Maharashtra, India SPG Consulting Full time

    About the RoleWe are seeking a highly skilled AI and Machine Learning Engineer to join our team. As an AI and Machine Learning Engineer, you will be responsible for designing and developing AI and machine learning models to solve complex business problems.Key ResponsibilitiesDesign and develop AI and machine learning models to solve complex business...


  • Mumbai, Maharashtra, India General Mills Full time

    At General Mills, we are passionate about creating a workplace that prioritizes being a force for good. As a Senior Machine Learning Engineer, you will play a critical role in driving this vision forward by developing and implementing cutting-edge AI and machine learning solutions that drive business growth and customer satisfaction.About the RoleThis is a...


  • Navi Mumbai, Maharashtra, India Reliance Industries Limited Full time

    About Us: {company} is a dynamic technology firm that thrives on innovation.Job Description: We are looking for a skilled Machine Learning Engineer to join our team. The ideal candidate will have experience in designing and developing machine learning models, as well as a strong understanding of software development principles.Design and develop machine...


  • Mumbai, Maharashtra, India Qrata Full time

    At Qrata, we're looking for a seasoned Senior Machine Learning Engineer to join our team. This role demands a high level of technical expertise and the ability to work independently on complex projects.Key ResponsibilitiesDevelop and optimize machine learning models to solve diverse and complex business problems.Collaborate closely with data scientists and...


  • Mumbai, Maharashtra, India Zycus Full time

    About ZycusZycus is a pioneer in Cognitive Procurement software, trusted by large global enterprises for two decades. Our Source to Pay integrated suite has been recognized by Gartner, Forrester, and other analysts. We power our S2P software with the revolutionary Merlin AI Suite, which takes over tactical tasks and empowers procurement and AP officers to...


  • Mumbai, Maharashtra, India Dew Software Full time

    Job Title: Machine Learning Engineer/SREWe are seeking a highly skilled Machine Learning Engineer/SRE to join our team at Dew Software. As a key member of our AI infrastructure team, you will be responsible for designing, implementing, and maintaining scalable and efficient machine learning models and infrastructure.Key Responsibilities:Azure Infrastructure...


  • Mumbai, Maharashtra, India CrowdStrike Full time

    CrowdStrike is a market leader in cybersecurity, and we're looking for talented engineers to join our Data + ML Platform team. As a Chief Machine Learning Engineer, you'll play a key role in building out our ML Experimentation Platform from the ground up.The ideal candidate will have 10+ years of experience developing and deploying machine learning solutions...


  • Mumbai, Maharashtra, India IDFC FIRST Bank Full time

    Key Responsibilities:The ideal candidate will be responsible for ensuring traceability and versioning of datasets, models, and evaluation pipelines.They will design, prototype, build, and maintain APIs for consumption of machine learning models at scale.Facilitating the development and deployment of proof-of-concept machine learning systems is also a key...


  • Mumbai, Maharashtra, India General Mills Full time

    About the RoleWe are seeking a highly skilled Lead Machine Learning Engineer to join our Global Data Science team at General Mills. In this role, you will be responsible for leading the development and deployment of machine learning solutions across the organization.As a Lead Machine Learning Engineer, you will work closely with cross-functional teams to...


  • Mumbai, Maharashtra, India IDfy Full time

    About the roleIn this position, you will be responsible for overseeing the development and maintenance of our production machine learning platform. This includes acquiring data, training and building models, deploying models, and creating API services for exposing these models.Key ResponsibilitiesWork on performance tuning of modelsFrom time to time, work on...


  • Mumbai, Maharashtra, India Cloud Counselage Pvt Ltd Full time

    At Cloud Counselage Pvt Ltd, we are seeking an experienced Senior Machine Learning Engineer to drive the development and deployment of cutting-edge machine learning models and algorithms. This role will play a key part in shaping our company's machine learning strategy and driving business impact through data-driven insights.Key Responsibilities:Design and...


  • Mumbai, Maharashtra, India AruGuru Tech Full time

    At AruGuru Tech, we are seeking a highly skilled Senior Embedded Machine Learning Engineer to contribute to the development of innovative machine learning solutions for embedded systems.This is a mid-level position requiring 4 to 6 years of experience in developing and implementing machine learning algorithms for embedded systems.The ideal candidate should...


  • Mumbai, Maharashtra, India Cloud Counselage Pvt Ltd Full time

    Role Overview:At Cloud Counselage Pvt Ltd, we are seeking a highly skilled Senior Cloud Machine Learning Engineer to lead our team in developing and deploying cutting-edge machine learning models and algorithms to solve complex business problems. As a Senior ML Engineer, you will work closely with cross-functional teams to design, implement, and optimize ML...