Sr. Software Engineer - ML Infrastructure
As a Software Engineer focusing on our machine learning infrastructure here at Baseten, you’ll build the infrastructure and services for our users to deploy, serve, and monitor their ML models.
AS AN ENGINEER, YOU
- have diverse experiences in machine learning from the infrastructure and operationalization perspective — in this role you generally won’t spend a ton of time training new models.
- understand core machine learning concepts and frameworks (such as PyTorch, Tensorflow, sklearn etc.).
- have experience with ML serving, and experience using GPUs for inference.
- demonstrate fluency with one or more general purpose programming languages, including but not limited to Python, Ruby, Javascript, or Go.
- have experience working (or desire to work with) with container-orchestration frameworks such as Kubernetes.
- have experience building scalable backends with modern web frameworks like Django, Rails, or Node.
- are comfortable designing and architecting systems and can balance performance, usability and effort requirements effectively.
- understand observability, SLA, and scalability implications, but are able to avoid premature optimization.
- have experience establishing standards and best-practices for other backend developers.
YOUR ALSO
- can own products and projects end-to-end; engineers and designers at Baseten also function as PMs, so we’d like everyone on the team to be able to empathize with our users, understand/write project specs, and manage the end-to-end execution of features.
- are comfortable with navigating ambiguity and enjoy the journey as much as the destination.
- are motivated by customer problems and find joy in creating simple, elegant solutions that avoid unnecessary complexity.
- exercise good judgment on tradeoffs and tools needed to solve the problem and don’t over index on trendy/fashionable tech unless it’s the right tool for the job.
- demonstrate pride, ownership, and accountability for your work and expect the same from your teammates.
OUR CURRENT TECH STACK
- Backend — Python + Django + Postgres + Redis
- Frontend — React + TypeScript + GraphQL
- Infrastructure — AWS + Kubernetes + Kubeflow + Knative + Prometheus
BENEFITS
- Competitive compensation. We aim to provide 90th percentile (or better) salaries and equity grants for every team member commensurate with their experience.
- Remote-first work environment. The Baseten team is welcome to work from wherever they want; fully remote, based in an office, or a mix of both. Today, our team (including our founding team) is spread across the United States, Canada, and Armenia. We have an office in downtown San Francisco. We provide a $1,000 stipend for you to make your home-office comfortable and productive.
- Quarterly in-person team summits. We get together as a team four times a year to plan, workshop, and most importantly, get to know each other better.
- Unlimited PTO. We ask that everyone take at least 4 weeks of vacation. And we have a company-wide break between Christmas and New Year's Day.
- 100% coverage of medical, dental and vision insurance for you and your family
- 16-weeks fully paid parental leave (adoptive and non-birth parents included) and flexibility with schedules while returning to work
- Company-sponsored 401(k) for you to contribute to
- Learning and development budget. We encourage you to take classes, attend conferences, and invest in your craft and we’ll cover expenses to make it happen