Senior Machine Learning Engineer

Senior Machine Learning Engineer

Job Overview

Company Name: Photoroom Location: Paris | Remote Department: General

About us: Photoroom launched in 2020 after being accepted into Y Combinator and has become the world's most popular AI photo editor over the past four years. Our goal is to create the technology allowing anyone to create studio-level product images in minutes. With over 300 million downloads and processing 5+ billion images annually, we serve both individual creators and major enterprises like Amazon, DoorDash, and Decathlon through our B2C app and B2B API solutions. We're a profitable, remote-friendly company that has raised Series B funding and aims for 40% year-over-year growth. Our team of 100+ passionate builders focuses on craft, innovation, and collaboration, creating exceptional impact for entrepreneurs and businesses worldwide.

Responsibilities

• Support the team in getting their models in the hands of millions of users as fast as possible with minimum work on their side while maintaining state-of-the-art inference performance.
• Own the ML CiCd and ensure that it is robust and reactive.
• Support researchers in speeding up their experiments through better compilation of training models, sharding strategies, and identifying bottlenecks.
• Collaborate within a multicultural team composed of ~100 passionate, friendly, and committed folks.
• Transmit your knowledge and experience through your code and reviews, prioritizing the whole team's progress and impact.

Requirements

• 4+ years of experience with PyTorch, TensorFlow, or Jax; willing to start or continue working with PyTorch.
• Experience in building CiCd pipelines that ensure code quality and consistency in both research and production codebases.
• Proven work in automating the model lifecycle from data collection, training, evaluation, deployment, and monitoring in production.
• Experience optimizing GPU power usage and model performance (latency, throughput, cost) in research or production environments.
• Track record of deploying several deep learning models in production using efficient runtimes in services used in real-world applications or products.
• Strong sense of ownership, initiative, and comfort in making product and technical decisions.
• Curiosity and ability to explain or learn depending on the situation, valuing knowledge sharing and humility within the team.
• Nice to have: Familiarity with TensorRT, Dynamo, Dynamo-Triton (ex Triton Inference Server), VLLM, TensorRT-LLM.

If you think you have what it takes but don't meet every single point above, please still apply. We'd love to chat and see if you could be a great fit.

Benefits

• Work flexibly from anywhere in Europe within a 3-hour flight from Paris (HQ).
• Regular team gatherings, including in-person onboarding in Paris (1-2 weeks), yearly company offsite and team retreat, quarterly in-person meetings (monthly during probation period), and social events like winter party and hackathons.
• 30 days annual leave plus local public holidays.
• Competitive equity package with stock options/BSPCE, giving you ownership in our growing company.
• €1000 one-time home office grant OR €400 per month co-working space stipend.
• €1, 000 annual learning and development budget for training, courses, books, and skill development.
• Private health insurance.
• Access to personalized mental health support, including 1:1 sessions with therapists or coaches, self-care tools, and well-being resources via MokaCare.
• Sports and cultural activities reimbursement.
• Relocation support (up to €10, 000) available for those choosing to move to France, including apartment finding and visa assistance.

Note: Visa sponsorship and relocation support are only available for positions based in France. We cannot sponsor visas for remote positions outside of France.

About the Company

Default company information, to be customized.