Skip to content
KreaKreaSan Francisco, CA

Machine Learning Engineer

Develops and trains large-scale diffusion models for image/video generation, controllability modules like IPAdapters/ControlNets, and novel research techniques for production. Requires proven experience with image/video models at scale and deep learning frameworks.

Salary not listed
On-siteML Engineering

About the role

Responsibilities

  • Train foundation diffusion models for image and video generation.
  • Train controllability modules such as IPAdapters or ControlNets.
  • Develop novel research techniques and put them into production.
  • Conduct large-scale experiments on high-performance computing clusters, optimizing data pipelines for massive image datasets.

Requirements

  • Proven track record in working with image or video models at scale (publications or open-source contributions a plus).
  • Strong background in deep learning frameworks and distributed training paradigms.
  • Ability to iterate rapidly, and propose creative research directions.

Skills

Diffusion ModelsIpadaptersControlnetsDeep LearningPyTorchTensorFlowDistributed TrainingImage GenerationVideo GenerationHigh-Performance Computing

Similar roles

ML Engineering jobs
OpenRouter

Applied AI Engineer

OpenRouterUnited States

Build and productionize internal agentic workflows and tooling on OpenRouter to automate support and go-to-market operations. Requires build-over-buy conviction, reliability focus, domain knowledge in support/GTM, backend systems expertise, security mindset, and quantitative evals.

Salary not listed
RemoteML Engineering
Beacon AI

Software Engineer, Artificial Intelligence/LLM

Beacon AISan Carlos, CA

Build and ship production LLM-powered features for aviation safety platform, including RAG, tool-calling, evals, guardrails, and monitoring for cost/latency/quality. Requires prior experience shipping LLM apps with strong RAG and production engineering skills.

135k – 260k/yr
Hybrid5+ YOEML Engineering
Beacon AI

Software Engineer, Cloud Infrastructure

Beacon AISan Carlos, CA

Build and operate scalable AWS cloud infrastructure, LLM platforms, RAG systems, and data pipelines for an AI aviation platform. Requires strong AWS and LLM production experience; hybrid in San Carlos, CA.

135k – 260k/yr
Hybrid5+ YOEML Engineering
Cloudflare

Software Engineer, AI Agents

CloudflareUnited States

Build and ship production AI agents on Cloudflare's edge platform using Workers, Durable Objects, and AI tools. Requires strong TypeScript/Rust experience, observability expertise, and hands-on LLM tooling for evals, safety, and multi-agent systems.

Salary not listed
On-siteML Engineering
Cerebras Systems

CoDesign & NextGen Performance Engineer

Cerebras SystemsSunnyvale, CA

Characterize, analyze, and optimize performance of state-of-the-art AI models on Cerebras' wafer-scale hardware. Build performance models, optimize kernels and compilers, debug runtime behavior, and develop visualization tools to influence next-gen AI architecture.

Salary not listed
On-site3+ YOEML Engineering