Roboflow
Des Moines, IA
11-50 employees
Making computer vision easy to use for developers. Roboflow creates software-as-a-service products to enable faster, easier computer vision.
Prior Listings
Full Stack Machine Learning Engineer
Full Stack Machine Learning Engineer
About the role
Roboflow is hiring full-time engineers on the machine learning team to contribute to the core machine learning tools and infrastructure that underpin the Roboflow platform.
This role will inevitably involve wearing a lot of hats. Wide-ranging curiosity and enthusiasm for diving into abstract problems, coming up with good solutions, and seeing them through to completion are essential responsibilities.
Our core belief is that computer vision is a foundational technology that is going to transform nearly every industry. This is an opportunity to shape how millions of developers will experience and use it for the first time. Your contribution will have a massive impact.
Skills - you should be familiar with many of these concepts and technologies and have built projects with some of them.
- Backend: node, Docker, python, Flask, pip, REST
- Machine Learning: PyTorch, TensorFlow, ONNX, OpenVINO, TensorRT, TFjs
- AWS: EC2, ECR, ECS, Lambda, Cloud Watch, Batch, S3
- Google Cloud: Cloud Functions, Cloud Storage, PubSub, GCE, GKE, Elastic
- Frontend: Firebase, React, jQuery, HTML, CSS
We anticipate the role being focused 60% on backend, 20% on machine learning, and 20% on other areas.
You certainly don't need to be an expert in all of these areas; but should be excited to learn new skill sets as you need them. The above tools span code in our machine learning stack to date - your opinions on new technologies that we adapt for new tasks will be highly valued.
What We Need from You
On the machine learning team, we primarily work on building and maintaining technology within Roboflow’s training, search and deployment services, but from time to time we're also helping deliver on enterprise contracts, and coding awesome open source projects and sample projects.
In the beginning, you will be tackling projects in close collaboration with your fellow machine learning and product team members. As you progress in your knowledge of Roboflow’s mission and tools, you will have a wide degree of freedom to advocate for and drive your own projects. If you need a rigid list of tasks spelled out in a multi-month roadmap, this role probably won't be a good fit.
We’re especially keen to add some rigor to our processes and build the foundation for scaling the engineering organization and the machine learning team.
Projects you might work on in your first few months:
- Improving and expanding the capabilities of the Roboflow Inference Server
- Building an ecosystem of add-ons that work closely with the Roboflow Inference Server to enable features like device tracking, active learning, etc.
- Ensuring uptime of training and inference infrastructure
- Iterating on training infrastructure to provide the best experience for our AutoML users while incurring the least cost for Roboflow
- Integrating Autodistill into the Roboflow platform to bootstrap projects with zero shot labeling.
Technology
Our goal is to build the world's best computer vision infrastructure so our users don't have to. This means we handle a lot of challenging complexities like seamlessly ingesting dozens of data formats, processing tens of millions of images per day, and deploying auto-scaling machine learning infrastructure that can handle our customers' most demanding training and deployment needs.
Our core app sits atop Firebase with assistance from auto-scaling groups of Docker containers (for jobs like archiving datasets and training models). We also heavily lean on serverless infrastructure so we can gracefully deal with bursty traffic involved in manipulating datasets that can range anywhere from one hundred to one million images.
Our machine learning infrastructure runs in AWS, with a few deployments spanning into GCP. We train and deploy various state of the art models in a variety of machine learning frameworks. All of our machine learning applications are closely integrated with the core Roboflow web application.
We also maintain a library of Colab notebooks our customers can use to train common computer vision models, a directory of public datasets, and a web of format specifications. We see building and supporting mini-projects like these that are helpful to the community at large as part of our role in democratizing computer vision.