Machine Learning Research Engineer, Senior

What you'll do

• Researches and develops machine-learning models, training methods, and algorithms across areas such as generative AI, reasoning, and reinforcement learning.
• Develops and optimizes models and software for efficient execution on edge devices and cloud platforms, balancing memory, speed, and power.
• Researches and implements machine-learning compiler components and compute approaches for hardware accelerators.

What they ask for

• Strong background in computer science and deep-learning fundamentals.
• Strong programming skills in Python and PyTorch, including designing and implementing complex training and evaluation pipelines.
• Exceptional analytical, development, debugging, and problem-solving skills.
• PhD in AI, Computer Science, Engineering, Information Systems, or a related field, or an MS with 4+ years of machine-learning research or related work experience.

Nice to have

• Experience with LLM, LVM, LMM models, and other neural network architectures.
• Experience with machine-learning accelerators, hardware optimization, heterogeneous or parallel computing systems, or on-device AI.

Pay

$140,800.00 - $211,200.00 USD base salary

Tools

Python, PyTorch, transformers, state-space networks, diffusion networks, VAEs

AI focus

Applied and fundamental machine-learning research, including model efficiency, generative AI, reasoning, reinforcement learning, on-device AI, and model/compiler optimization for accelerators.