AI/ML Job: Machine Learning Software Engineer

Machine Learning Software Engineer at Shield AI

🇺🇸 United States › California › San Francisco   (Posted Dec 12 2021)

About the company
Shield AI’s mission is to protect service members and civilians with artificially intelligent systems. We currently design, build and deploy a quadrotor robot that autonomously explores potentially hazardous environments to help service members make informed decisions. For our world-class team, no idea is too ambitious, and we never stop working to make possible what looks out of reach today. We are backed by Silicon Valley venture capital firms including Andreessen Horowitz, have been shipping product since 2018, and are growing rapidly.

Job description
Help us build world-class infrastructure for training, evaluating and productizing machine learning models! As a member of our machine learning software infrastructure team you will work with colleagues to improve workflows, develop pipelines to train and evaluate advanced machine learning models and ensure these pipelines are efficient and well-tested. At Shield you’ll be part of an elite team that uses machine learning to enable autonomous robots that save lives.

What you'll do:
Work with colleagues to design, build and maintain high performance software libraries and pipelines for machine learning.
Work with team members to deliver software features on-time and meeting requirements.
Identify performance bottlenecks and design and implement solutions for them.
Improve visibility into model training and experiment tracking for machine learning engineers.
Document code and workflows to enable engineers to rapidly implement solutions using the pipelines.
Maintain a small machine learning model for demonstration, prototyping, and testing.
Help manage our cluster resources.
Learn and apply best practices for state of the art machine learning development.

Projects you might work on:
Setup model tracking and metric visualizations using jupyter notebooks, mlflow, tensorboard or other modern tools.
Optimize pipeline implementations to reduce training time and maximize compute resource utilization.
Enable efficient hyperparameter optimization.

Required qualifications:
Bachelors in computer science, software engineering, machine learning engineering or similar field.
You have a demonstrated record of working hard, being a trustworthy teammate, holding yourself and others to high standards, and being kind to others.
Good written and verbal communication skills.

Apply now!