Machine Learning Engineer

About The Company

Roboflow is dedicated to transforming the world into a programmable environment through advancements in artificial intelligence and computer vision. Our mission is to simplify building and deploying machine learning models, making sophisticated vision capabilities accessible to developers worldwide. With over 1 million users, including half of the Fortune 100 companies, Roboflow supports a diverse array of applications such as medical research, construction safety, digital mapping, environmental conservation, drone navigation, and more. Our innovative tools, open-source projects, and hosted solutions empower developers to accelerate their projects and bring cutting-edge AI applications to life.

Supported by prominent investors including Y Combinator, Google Ventures, and Craft Ventures, Roboflow has raised over $63 million to fuel our growth and innovation. Our team values ownership, accountability, and a proactive approach to problem-solving. We foster a high-autonomy environment where curiosity, technical passion, and a founder mindset are highly regarded. Our culture encourages continuous learning, collaboration, and building impactful solutions with passionate teammates committed to shaping the future of computer vision technology.

About The Role

As a Machine Learning Engineer at Roboflow, you will play a vital role in deploying and optimizing state-of-the-art computer vision models for production environments. This is a highly execution-focused position, emphasizing the practical aspects of deploying scalable, high-performance machine learning services rather than conducting research. Your work will directly impact thousands of developers by enabling them to leverage advanced vision models in real-world applications. You will collaborate closely with cross-functional teams—including product, infrastructure, and research—to integrate new model capabilities, enhance inference performance, and ensure system robustness and maintainability.

Your responsibilities will include optimizing and deploying machine learning models using tools like TensorRT and PyTorch, building and maintaining inference services, and monitoring their performance at scale. You will also work on integrating new models into training and inference pipelines, developing analytics pipelines for large-scale vision data, and exposing new model features via APIs and user interfaces. Troubleshooting full-stack issues in Python and JavaScript will be part of your routine, ensuring seamless operation of our machine learning services.

Qualifications

• 4–5 years of professional experience in Python programming
• 3+ years of experience in machine learning and deploying ML models in production environments
• Strong hands-on skills with PyTorch, TensorRT, and Python
• Proficiency in JavaScript for full-stack development and debugging
• Experience with system architecture, including NoSQL and SQL databases
• Familiarity with serverless computing environments
• Knowledge of computer vision techniques such as object detection, classification, segmentation, YOLO, VLMs, and SAM
• Demonstrated ability to write maintainable, modular, and scalable code following engineering best practices
• Experience working in startup or high-ownership roles, with the ability to operate autonomously in a fast-paced environment
• Strong problem-solving skills and a focus on delivering value to end-users

Responsibilities

• Optimize and deploy machine learning models for performance and scalability using tools like TensorRT and PyTorch
• Build, maintain, and monitor inference services for vision models in production
• Integrate new models into training and inference pipelines, ensuring robustness and efficiency
• Develop analytics pipelines capable of handling large-scale vision data, including billions of images
• Collaborate with research and product teams to expose new model capabilities via APIs and user interfaces
• Diagnose and debug full-stack issues across ML services using Python and JavaScript
• Contribute to system architecture decisions and improve deployment workflows for machine learning models
• Stay updated with the latest advancements in computer vision and incorporate them into production systems

Benefits

• Target annual salary range of $200,000 to $275,000, reviewed biannually to stay competitive
• Equity package to ensure shared success and future growth
• $4,000 yearly travel stipend for flexible work locations
• $350 monthly productivity stipend to enhance your work environment
• $350 monthly AI tools budget to access the latest LLM technologies
• Coverage of up to 100% health insurance costs for you and your family or partner
• Remote-first and flexible scheduling to promote work-life balance
• Unlimited paid time off with a minimum of two weeks annually
• 12 weeks parental leave to support family growth

Equal Opportunity

At Roboflow, we believe that great ideas come from diverse perspectives and backgrounds. We are committed to building an inclusive environment and are proud to be an Equal Opportunity Employer. We consider all qualified applicants regardless of race, color, religion, sex, sexual orientation, gender identity, national origin, disability, age, veteran status, or any other legally protected characteristic.