Software Engineer - Machine Learning
FetchFull Time
Mid-level (3 to 4 years), Senior (5 to 8 years)
Candidates should have 4-5 years of professional experience in Python, with at least 3 years focused on machine learning and production deployment of ML models. Strong hands-on skills with PyTorch, TensorRT, and Python are required, along with proficiency in JavaScript. Familiarity with object detection, classification, segmentation, YOLO, VLMs, and SAM is also necessary. Experience working in startups or high-ownership roles, and comfort with autonomy, ambiguity, and rapidly shifting priorities are essential.
The Machine Learning Engineer will optimize and deploy production ML models for performance and scalability using tools like TensorRT and PyTorch. They will build, maintain, and monitor inference services for vision models, integrate new models into training and inference pipelines, and construct analytics pipelines for large-scale vision data. The role involves collaborating with researchers and engineers to expose new model capabilities via APIs and UIs, and diagnosing full-stack issues in Python and JavaScript across ML services.
Platform for creating and deploying AI models
Roboflow offers a platform for engineers to create, train, and deploy machine learning models using their own images and videos. The platform features an auto-annotate API for efficient data labeling, along with tools for preprocessing and augmenting image data. Roboflow distinguishes itself from competitors by providing project management tools that enhance team collaboration on AI projects. The company's goal is to simplify the AI development process for a diverse range of clients, from individual engineers to large organizations.