Skip to content

Software Engineer, ML Platform

Builds and owns production ML platform systems, turning AI research into reliable, scalable features. Partners with CTO on prototyping, observability, and end-to-end deployment of AI capabilities. Requires 3+ years experience with Python and modern ML frameworks; onsite in NYC.

150k – 350kNew York, NYML EngineeringOnsite3+ YOE

About the role

What you'll do

  • Partner directly with our CTO to turn AI research and experiments into production-grade features
  • Investigate and prototype new AI capabilities, taking early-stage ideas and quickly determining what's feasible to build and ship
  • Build observability and evals infrastructure so we can monitor models in production and measure whether they're succeeding
  • Take ownership of AI features end-to-end, from design through deployment and monitoring, collaborating with other teams as needed to bring features to production

What we're looking for

  • 3+ years of experience as an engineer, including recent work on cutting-edge ML and AI systems
  • Owned production AI features end-to-end, from experimentation to maintaining live production infrastructure
  • Proficient in Python with hands-on experience in modern ML frameworks (e.g. PyTorch, HuggingFace, DSPy)
  • Open to working 5 days a week out of our office in NYC

You might excel if

  • Hands-on experience operating at an early stage startup where ownership and execution in the face of ambiguity were essential
  • ML research background on top of engineering skills, particularly in areas relevant to Sola (e.g. CUA, agentic reasoning, visual grounding, VLMs, fine-tuning)
  • Expert in some or all of our stack: Typescript, Python, DSPy, vLLM / TensorRT, Temporal, Kubernetes

Compensation & Benefits

  • Competitive salary plus meaningful equity at Series A stage
  • Comprehensive medical, dental, and vision benefits
  • 401(k) matching
  • Daily meals together in the office
  • Flexible PTO (4 weeks per year)
  • Relocation support for candidates not already based in NYC

Skills

PythonPyTorchHuggingfaceDspyvLLMTensorRTTemporalKubernetesTypeScriptMl Frameworks

Similar roles

ML Engineering jobs

AI Engineer

Build full-stack AI prototypes and agentic systems to pressure-test venture ideas. Requires 3+ years building production AI applications with strong frontend/backend fluency and frontier coding agent expertise.

150k – 190kMountain View, CAML EngineeringOn-site3+ YOESQLAPIs

Machine Learning Engineer

Build and deploy ML models for entity resolution and knowledge graph expansion on large-scale China-related data. Requires 4+ years clustering ML experience and end-to-end production ML with Python/SQL.

150k – 195kNew York, NYML EngineeringHybrid4+ YOESQLNLP

Research Engineer, Post-Training

Research Engineers design and run post-training workflows, build evaluation infrastructure, and turn frontier AI techniques into reliable production systems for enterprise customers. Requires experience with fine-tuning, RLHF, reward modeling, and strong experimentation skills.

150k – 250kSan Francisco, CA +1ML EngineeringHybridEvalsPython

Research Engineer, Agents

Research Engineers build and productionize agentic AI systems, designing compound architectures, evaluation frameworks, and reliable execution for enterprise workflows. Requires strong Python skills, systems reasoning, and experience building agents with tools, retrieval, planning, and memory.

150k – 250kSan Francisco, CA +1ML EngineeringHybridPythonMemory

ML Engineer

Build and deploy production ML models and pipelines to detect suspicious activity, improve verification accuracy, and support threat intelligence workflows.

150k – 180kUnited StatesML EngineeringRemote4+ YOEAWSClustering