AI Model Optimization Expert

4 days ago


Bengaluru, Karnataka, India beBeeArtificialIntelligence Full time ₹ 15,00,000 - ₹ 28,00,000
AI Model Optimization Specialist

Job Description:

We are seeking a highly skilled AI model optimization specialist to join our team. In this role, you will be responsible for enabling and optimizing key AI models on AMD GPUs, collaborating with internal GPU library teams, and working with open-source framework maintainers to integrate your code changes upstream.

Responsibilities:

  • Enable and optimize key AI models (LLM, Vision, MultiModal, etc.) on AMD GPUs
  • Optimize AI frameworks like PyTorch, TensorFlow, etc. on AMD GPUs in upstream open-source repositories
  • Collaborate and interact with internal GPU library teams to analyze and optimize training and inference for AI
  • Work with open-source framework maintainers to understand their requirements and have your code changes integrated upstream
  • Optimize GPU kernels, understand and drive AI operator performance (GEMM, Attention, etc.) with specialized teams
  • Work in a distributed computing setting to optimize for both scale-up (multi-GPU) and scale-out (multi-node) systems
  • Apply your knowledge of software engineering best practices

Required Skills and Qualifications:

  • Strong technical and analytical skills in C++/Python AI development in a Linux environment
  • Ability to work as part of a team, while also being able to work independently, define goals and scope, and lead your own development effort
  • Knowledge of GPU computing (HIP, CUDA, OpenCL)
  • AI model experience or knowledge - Natural Language Processing, Vision, Audio, Recommendation systems
  • Excellent C/C++/Python programming and software design skills, including debugging, performance analysis, and test design
  • Experiences to run workloads on large scale heterogeneous cluster is a plus
  • Experiences to optimize GPU kernels for performance is a plus

Academic Credentials:

  • Masters or PhD or equivalent experience in Computer Science, Computer Engineering, or related field

  • AI Model Optimizer

    3 days ago


    Bengaluru, Karnataka, India beBeeData Full time ₹ 1,20,00,000 - ₹ 2,00,00,000

    About ArctanWe are solving a hard problem that matters: Eliminating communication barriers in global conversations. Our AI product refines speech on the fly, adjusting accents and tonality, helping customer-facing teams engage better with their customers and build rapport effortlessly across cultures.Founded in 2024 by IIT Roorkee alumni, we are backed by...


  • Bengaluru, Karnataka, India beBeeMachineLearning Full time US$ 1,50,000 - US$ 2,50,000

    Job OverviewAs an expert in artificial intelligence and machine learning, you will play a pivotal role in developing innovative solutions for edge platforms.Key Responsibilities:Optimize Gen AI models for edge platforms through quantization, pruning, distillation, and finetuning.Develop fundamental understanding of Amazon's Neural Edge Engine to invent...


  • Bengaluru, Karnataka, India beBeeArtificialIntelligenceEngineer Full time ₹ 15,00,000 - ₹ 25,00,000

    Job Title: Artificial Intelligence Engineer">">As an AI engineer, you will play a key role in designing, developing and optimizing machine learning models for classification, regression and recommendation systems.Key Responsibilities:">Design, develop and optimize machine learning and AI models for classification, regression and recommendation systems;Build...


  • Bengaluru, Karnataka, India beBeeMachineLearning Full time ₹ 20,00,000 - ₹ 25,00,000

    AI Model Optimization SpecialistWe are seeking a seasoned AI/ML professional to develop and deploy cutting-edge models on edge devices.The ideal candidate will have a strong foundation in machine learning, deep learning, and hands-on experience with Python frameworks such as TensorFlow and PyTorch.Key Responsibilities:Design and implement anomaly detection...


  • Bengaluru, Karnataka, India Awign Expert Full time US$ 1,20,000 - US$ 2,00,000 per year

    We are seeking a highly skilled AI/ML Engineer with expertise in AWS AI/ML services and a strong understanding of Generative AI using Amazon Bedrock. The ideal candidate will have experience in building, deploying, and optimizing AI/ML models on AWS, integrating LLMs into applications, and leveraging AWS services for scalable AI solutions.Key...


  • Bengaluru, Karnataka, India beBeeArtificialIntelligence Full time ₹ 1,20,00,000 - ₹ 2,00,00,000

    Optimize AI on the EdgeWe are seeking exceptional scientists to develop state-of-the-art techniques for General Artificial Intelligence (Gen AI) on edge platforms. This involves quantizing, pruning, distilling, and fine-tuning Gen AI models to optimize performance.Fundamental understanding of Amazon's Neural Edge Engine is necessary to invent optimization...

  • AI Architect

    6 days ago


    Bengaluru, Karnataka, India Quickhyre AI Full time

    We are seeking an experienced AI Architect to lead the design, development, and deployment of large-scale AI solutions. The ideal candidate will bridge the gap between business requirements and technical implementation, with deep expertise in generative AI and modern MLOps practices.Key ResponsibilitiesAI Solution Design & ImplementationArchitect end-to-end...


  • Bengaluru, Karnataka, India beBeeMachineLearning Full time ₹ 2,00,00,000 - ₹ 2,50,00,000

    Job Title: Senior Machine Learning EngineerAbout the Role:We are seeking a highly skilled and experienced Senior Machine Learning Engineer to join our team. As a key member of our engineering team, you will be responsible for designing, developing, and deploying cutting-edge machine learning models that drive business value.Key Responsibilities:Design and...


  • Bengaluru, Karnataka, India beBeeDeveloper Full time ₹ 1,04,000 - ₹ 1,30,878

    Job Title:AI Model DeveloperWe are seeking an experienced AI Model Developer to join our team. The successful candidate will be responsible for building and maintaining Generative AI models using cloud-based services.The ideal candidate will have a strong understanding of cloud infrastructure and experience with Generative AI models. They will also be...


  • Bengaluru, Karnataka, India beBeeMachineLearning Full time ₹ 15,00,000 - ₹ 25,00,000

    Role OverviewWe are seeking a skilled Machine Learning Engineer to join our team. The ideal candidate will have a strong background in machine learning, data manipulation, and cloud platforms.Key ResponsibilitiesManage and optimize end-to-end MLOps pipelines for data collection model training, validation, and monitoring while ensuring team collaboration and...