Principal ML Engineer

6 days ago

Bengaluru, Karnataka, India AMD Full time ₹ 20,00,000 - ₹ 25,00,000 per year

WHAT YOU DO AT AMD CHANGES EVERYTHING
We care deeply about transforming lives with AMD technology to enrich our industry, our communities, and the world. Our mission is to build great products that accelerate next-generation computing experiences - the building blocks for the data center, artificial intelligence, PCs, gaming and embedded. Underpinning our mission is the AMD culture. We push the limits of innovation to solve the world's most important challenges. We strive for execution excellence while being direct, humble, collaborative, and inclusive of diverse perspectives.

AMD together we advance_

Principal ML ENGINEER
The Role

We are looking for Machine Learning Engineer to join our Models and Applications team. If the challenge of distributed training of large model on large number of GPUs excites you and you are passionate about improving training efficiency and enjoy innovating and coming up with new ideas, then this role is for you.

You will be part of world class team focus on addressing the challenge of training generative AI.

The Person

The ideal candidate should have experience with distributed training pipeline, knowledgeable with distributed training algorithms (Data parallel, Tensor parallel, Pipeline parallel, ZeRO) and familiar with training Large Model.

Key Responsibilities

  • Train large model to convergence on AMD GPUs.
  • Improve the end-to-end training pipeline performance.
  • Optimize the distributed training pipeline and algorithm to scale out.
  • Contribute your changes to open source.
  • Up to date with latest training algorithms.
  • Influence the direction of AMD AI platform.
  • Cross team collaborate with various group and stakeholder.

Preferred Experience

  • 10+ years of experience.
  • Experience in ML frameworks such as PyTorch, JAX or Tensorflow.
  • Experience with distributed training and distributed training framework such as DeepSpeed.
  • Experience with LLM or Vision, especially large model is a plus.
  • Excellent python programing skills, including debugging, profiling, and perf analysis.
  • Experience with ML pipeline.
  • Strong communication and problem-solving skills.

Academic Credentials

A master's degree in computer science, artificial intelligence, machine learning, or a related field.

Benefits offered are described:
AMD benefits at a glance.

AMD does not accept unsolicited resumes from headhunters, recruitment agencies, or fee-based recruitment services. AMD and its subsidiaries are equal opportunity, inclusive employers and will consider all applicants without regard to age, ancestry, color, marital status, medical condition, mental or physical disability, national origin, race, religion, political and/or third-party affiliation, sex, pregnancy, sexual orientation, gender identity, military or veteran status, or any other characteristic protected by law. We encourage applications from all qualified candidates and will accommodate applicants' needs under the respective laws throughout all stages of the recruitment and selection process.

  • Principal - AI/ML Engineer

    6 days ago

    Bengaluru, Karnataka, India Fidelity Investments Full time ₹ 12,00,000 - ₹ 36,00,000 per year

    Job Description:Job Title –Principal – ML EngineerThe Purpose of This RoleWe are looking for a highly experienced and technically proficient Principal ML Engineer to lead the development, deployment, and scaling of machine learning infrastructure and pipelines and ML based applications. This role is ideal for someone with a strong foundation in backend...

  • Principal AI Engineer

    2 weeks ago

    Bengaluru, Karnataka, India Ecolab Global Services Full time ₹ 20,00,000 - ₹ 25,00,000 per year

    The Principal Data Scientist/AI Engineer is a senior technical expert who leads complex, high-impact AI initiatives across the organization. This role is responsible for driving architectural decisions, establishing reusable frameworks, and influencing the AI strategy at an enterprise level. The Principal AI Engineer operates as a cross-functional advisor,...

  • Principal/Sr. Principal-Data Scientist

    1 week ago

    Bengaluru, Karnataka, India Symmetrical Global Search Full time ₹ 20,00,000 - ₹ 25,00,000 per year

    We are Hiring For Principal/Sr. Principal-Data ScientistExperience: 9+ yearsJob descriptionDevelop, Test, and Deploy ML/AI models for various productsPerform data preprocessing, feature engineering and ML/DL model evaluationOptimize and fine-tune models for performance and scalabilityGood understanding of Statistical, ML, AI modelsGood understanding of NLP...

  • Principal Software Engineer

    3 weeks ago

    Bengaluru, Karnataka, India NIKE Full time

    PRINCIPAL SITE RELIABILITY ENGINEERIndia Technology Center WHO YOU WILL WORK WITHThe Principal Site Reliability Engineer will work alongside a talented team of Site Reliability Engineers focused on delivering reliabile and observable software used by millions of athletes* around the world.  You will be a part of the Resilience Engineering organization which...

  • Principal Software Engineer

    1 week ago

    Bengaluru, Karnataka, India NIKE Full time ₹ 12,00,000 - ₹ 36,00,000 per year

    PRINCIPAL SITE RELIABILITY ENGINEERIndia Technology CenterWHO YOU WILL WORK WITHThe Principal Site Reliability Engineer will work alongside a talented team of Site Reliability Engineers focused on delivering reliabile and observable software used by millions of athletes* around the world. You will be a part of the Resilience Engineering organization which...

  • Principal Support Engineer

    14 hours ago

    Bengaluru, Karnataka, India Oracle Financial Services Software Ltd Full time ₹ 20,00,000 - ₹ 25,00,000 per year

    Principal Software Engineer As a member of the Support organization, your role is to provide post-sales support and solutions to Oracle's customer base, acting as an advocate for customer needs. This includes addressing non-technical inquiries via phone and electronic channels, as well as responding to technical questions related to the use and...

  • Principal Engineer, CSS Global SaaS

    4 days ago

    Bengaluru, Karnataka, India Oracle Financial Services Software Ltd Full time ₹ 20,00,000 - ₹ 25,00,000 per year

    Principal Engineer, CSS Global SaaS & Apps Delivery, DevOps We are looking for a versatile Senior Developer & Database Engineer with expertise across AI/ML, cloud-native development, DevOps, and enterprise database administration. In this role, you will work on building and optimizing AI/ML pipelines, developing scalable cloud-based applications, and...

  • Principal Machine Learning Scientist

    3 weeks ago

    Bengaluru, Karnataka, India Vito Full time

    Role : Principal Machine Learning Scientist AdTechLocation : Bangalore (5 days work from office)Experience : 1015 yearsAbout the Role : We are hiring a Principal Machine Learning Scientist to lead large-scale ML initiatives in the AdTech domain. This role involves developing and deploying advanced algorithms in personalization, ranking, targeting, and...

  • Principal AI/ML Field Applications Engineer

    6 days ago

    Bengaluru, Karnataka, India SiMa Full time ₹ 12,00,000 - ₹ 36,00,000 per year

    DescriptionJob Title: Principal AI/ML Field Applications EngineerJob Location: Bangalore, India(This position requires a full-time, on-site presence in our Bangalore, India office)Job ID: AI2431Job DescriptionKey Responsibilities:Engage with India customers and Partners for enabling them to design Physical AI solutions developed with platform.Guide and...

  • Principal Engineer

    4 weeks ago

    Bengaluru, Karnataka, India Licious Full time

    About us With our de-licious fresh meat and seafood, we are Licious —India's leading D2C food-tech brand. Founded in 2015 by Abhay Hanjura and Vivek Gupta , we are headquartered in Bengaluru, delivering high-quality, fresh meat and seafood to over 32 lakh customers across 20 cities. Ready to add your unique flavor to Licious? Read on Role Overview We're...