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.
About the job
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
Python, PyTorch, Huggingface, Dspy, vLLM, TensorRT, Temporal, Kubernetes, TypeScript, Ml Frameworks
Similar jobs
ML Engineering jobs
LangChain
New York, NYBuild and teach reliable AI agent systems through customer workshops, technical content, guidance, and reference implementations. The role requires strong Python and agent-development experience plus a background delivering customer-facing technical training.
Roboflow
San Francisco, CA
Build reinforcement-learning environments, evaluations, datasets, and scalable infrastructure for frontier AI capabilities. The role suits a high-agency generalist engineer with experience in agents, evaluations, or RL workflows and strong communication skills.
Beacon Biosignals
Boston, MADevelop and productionize machine- and deep-learning algorithms for biosignal and EEG data used in medical devices, clinical development, and diagnostics. The role requires 4+ years of industry experience, DSP and statistics expertise, PyTorch proficiency, and familiarity with regulated environments and production ML practices.
Applied Intuition
Sunnyvale, CA
Develop and deploy ML-first behavior prediction and planning systems for autonomous vehicles, forecasting the motion and interactions of road users. Requires a bachelor's degree, deep learning lifecycle expertise, and at least three years of production software experience with C++ or Python.
Fab2
Austin, TXBuild the AI platform behind fab2, including model infrastructure, agent systems, evaluations, and tools for engineering and fab operations. The role requires strong production software engineering skills and comfort working across frontend, backend, infrastructure, and data.