✦ Luna Orbit — AI & Machine Learning

Sr. Staff Engineer, AI Models and Applications

at Advanced Micro Devices

📍 San Jose, California, United States Hybrid Posted March 13, 2026
Type Not Specified
Experience senior
Exp. Years Not specified
Education Not specified
Category AI & Machine Learning

This role focuses on enabling scalable and efficient training and inference for Generative AI models, including large language and multimodal models, leveraging state-of-the-art algorithms and frameworks.

  • Develop scalable training/inference techniques
  • Optimize generative AI models
  • Collaborate on software frameworks
  • Innovate low-precision training
  • Work with open-source AI community

The technical environment includes Generative AI architectures, large language models, distributed training techniques, and frameworks like PyTorch, JAX, and Hugging Face, optimized for AMD hardware.

The ideal candidate is a senior AI engineer with expertise in generative AI, large language models, and scalable training/inference techniques. They are passionate about innovating in AI model training and inference at scale, with experience working on cutting-edge frameworks like PyTorch, JAX, and Hugging Face.

Generative AILarge Language Modelsmodel trainingmodel inferencedistributed traininglow-precision trainingPyTorchJAXHugging Face
state-of-the-art GenAI algorithmsscaling traininginference optimizationsoftware-hardware co-optimization
PyTorchJAXHugging FaceAMD hardware
Generative AILarge Language ModelsLLMsMultimodal ModelsLMMsmodel trainingmodel inferencedistributed traininglow-precision trainingPyTorchJAXHugging Face
Generative AILarge Language ModelsLLMsMultimodal ModelsLMMsModel trainingModel inferenceDistributed trainingLow-precision trainingPyTorchJAXHugging FaceAI model optimizationscaling AI modelssoftware frameworks
innovative thinkingcollaborationproblem-solvingpassion for AIcommunication
Industry Technology
Job Function Generative AI Model Training and Optimization
Generative AILarge Language ModelsLLMsMultimodal ModelsLMMsmodel trainingmodel inferencedistributed traininglow-precision trainingPyTorchJAXHugging FaceAI model optimizationscaling AI models

Lack of experience with generative AI or LLMs, No experience with model training/inference at scale, Unable to work with PyTorch, JAX, or Hugging Face

Apply for this Position →

Get matched to jobs like this

Luna finds roles that fit your skills and career goals — no endless scrolling required.

Create a Free Profile