AI Research Engineer

You will design, build, and iterate on machine-learning pipelines for foundational models. You will train and evaluate transformer, diffusion, and reinforcement-learning models, turn research ideas into working code, integrate ML components into real-world systems, and share AI research insights.

Responsibilities

• Design, build, and iterate on experimental ML pipelines
• Implement and train transformer and diffusion-based models
• Develop and evaluate reinforcement-learning algorithms and frameworks
• Prototype and deploy research ideas into working code
• Integrate ML components into real-world systems
• Share AI research insights internally and externally

Requirements

• Master’s degree in Machine Learning, AI, Computer Science, or a related technical field
• At least 3 years of applied ML research and software engineering experience
• Experience training large-scale ML models such as transformers or diffusion models
• Understanding of reinforcement learning
• Python proficiency
• Experience with PyTorch, JAX, or TensorFlow
• Ability to write clean, efficient, scalable code
• Collaboration and communication skills
• US work authorization

Benefits

• Relocation stipend for eligible candidates relocating to the Seattle office