Skip to content
Raspberry AIRaspberry AIUnited States

Senior Machine Learning Engineer

Develops and optimizes diffusion models and generative AI for fashion design software. Conducts research, implements techniques for design quality and controllability, and deploys to production. Requires Master's/PhD, 3+ years experience, PyTorch proficiency.

Salary not listed
Remote3+ YOEML Engineering

About the role

Responsibilities

  • Conduct applied research and experimentation on state-of-the-art diffusion model architectures and training techniques.
  • Implement and evaluate novel techniques for improving quality and controllability in generated designs.
  • Analyze and interpret experimental results, draw meaningful conclusions, and communicate findings effectively.
  • Collaborate closely with the team to translate prototypes into production-ready systems.
  • Stay abreast of the latest advancements in diffusion models, deep learning, and generative AI research.

Requirements

  • Master's or Ph.D. in Computer Science, Machine Learning, or a related field.
  • 3+ years of industry experience.
  • Strong theoretical and practical understanding of deep learning, with a focus on generative models (e.g., GANs, VAEs, Diffusion Models).
  • Hands-on experience with deep learning frameworks such as PyTorch.
  • Experience with training and evaluating generative models on cloud GPU platforms (e.g., AWS, GCP, Azure).
  • Proficiency in using and tuning multimodal LLMs, including experience with both API-based and open-source model implementations.
  • Ability to effectively present complex technical information to both technical and non-technical audiences.

Skills

Diffusion ModelsPyTorchGansVaesAWSGCPAzureMultimodal LlmsDeep LearningGenerative AI

Similar roles

ML Engineering jobs
Scale AI

Senior Machine Learning Engineer, Agent Oversight

Scale AISan Francisco, CA +1

Senior ML Engineer building observability, evaluation frameworks, and improvement loops for production agentic AI systems. Requires 5+ years production ML/LLM experience, strong grounding in agent design or evaluation, and hands-on work taking systems from prototype to scale.

216k – 270k/yr
On-site5+ YOEML Engineering
Scale AI

Senior Software Engineer, Agent Oversight

Scale AISan Francisco, CA +1

Build platform infrastructure for observing, evaluating, and improving production AI agents at scale. Requires 4+ years software engineering experience with ML/LLM systems, strong backend/distributed systems skills, and collaboration with ML engineers.

216k – 270k/yr
On-site5+ YOEML Engineering
Ambient.ai

Senior Software Engineer, AI Infrastructure

Ambient.aiRedwood City, CA

Build and optimize scalable AI infrastructure for real-time inference, evaluation, and continuous improvement of LLMs, LVMs, computer vision, and multimodal models on large-scale video data. Requires 4+ years production ML systems experience, strong Python skills, and expertise in inference optimization and model serving.

168k – 205k/yr
Hybrid4+ YOEML Engineering
SpotOn

Senior Software Engineer

SpotOnChicago, IL +3

Senior AI Engineer building agentic workflows and orchestration layers to automate manual business processes for operations, sales, and customer support teams. Requires 7+ years software engineering experience, production LLM/agent expertise, and strong Python/TypeScript skills.

160k – 190k/yr
Hybrid7+ YOEML Engineering
Snowflake

Senior AI Engineer, Enterprise

SnowflakeMenlo Park, CA

Senior AI Engineer building scalable backend services, distributed systems, and reusable AI platform components (agent frameworks, evaluation pipelines, SDKs) that power Snowflake's internal agentic applications for GTM teams. Requires 5+ years software engineering experience, strong Python/FastAPI skills, and production LLM experience.

156k – 224k/yr
On-site5+ YOEML Engineering