Software Engineer, Machine Learning Infrastructure

Posted 3 Hours Ago
Be an Early Applicant
Redwood City, CA
180K-250K Annually
Senior level
Artificial Intelligence • Software
The Role
Design, build, and maintain training infrastructure for ML research. Collaborate with teams on product features and ensure system reliability.
Summary Generated by Built In

About the Company

Companies want to train their own large models on their own data. The current industry standard is to train on a random sample of your data, which is inefficient at best and actively harmful to model quality at worst. There is compelling research showing that smarter data selection can train better models faster—we know because we did much of this research. Given the high costs of training, this presents a huge market opportunity. We founded DatologyAI to translate this research into tools that enable enterprise customers to identify the right data on which to train, resulting in better models for cheaper. Our team has pioneered deep learning data research, built startups, and created tools for enterprise ML. For more details, check out our recent blog posts sharing our high-level results for text models and image-text models.

We've raised over $57M in funding from top investors like Radical Ventures, Amplify Partners, Felicis, Microsoft, Amazon, and notable angels like Jeff Dean, Geoff Hinton, Yann LeCun and Elad Gil. We're rapidly scaling our team and computing resources to revolutionize data curation across modalities.

This role is based in Redwood City, CA. We are in office 4 days a week.

About the Role

We’re looking for seasoned ML Infrastructure engineers with experience designing, building, and maintaining training infrastructure for our in-house ML research and validation efforts and the core infrastructure for running the curation pipeline that we deliver to our customers. As one of our early senior hires, you will partner closely with our founders on the direction of our product and drive business-critical technical decisions.

You will contribute to developing core infrastructure components that impact our ability to deliver, scale, and deploy our product. These are key components of our stack that allow us to process customer data and apply state-of-the-art research to identify the most informative data points in large-scale datasets. You will have a broad impact on the technology, product, and our company's culture.

What You'll Work On

  • Architect, build and maintain the infrastructure that ensures highly available GPU workloads for training-purposes

  • Troubleshoot and resolve issues across GPU resources, networking, OS, drivers, and cloud environments, automate detection and recovery of such issues

  • Design, build, and maintain the infrastructure that powers our data curation product.

  • Partner with researchers and engineers to bring new features and research capabilities to our customers

  • Ensure that our infrastructure and systems are reliable, secure, and worthy of our customers' trust.

About You

There are a few specific things we’ll be looking for that will help you succeed in this role:

  • 5+ years of experience

  • Have meaningful experience with leading and building production ML infrastructure and platforms that deliver on major product initiatives.

  • Proficiency in Python and in the most commonly used tools in the infrastructure space: Linux, Kubernetes, Terraform / Pulumi, etc

  • Strong knowledge of hardening cloud native and especially K8s workloads.

  • Experience maintaining a high-quality bar for design, correctness, and testing.

  • Have a humble attitude, an eagerness to help your colleagues, and a desire to do whatever it takes to make the team succeed

  • Own problems end-to-end and are willing to pick up whatever knowledge you're missing to get the job done.

  • Experience running data-processing workloads in k8s (e.g spark on k8s)

Compensation

At DatologyAI, we are dedicated to rewarding talent with highly competitive salary and significant equity. The salary for this position ranges from $180,000 to $250,000.

  • The candidate's starting pay will be determined based on job-related skills, experience, qualifications, and interview performance.

Benefits

We offer a comprehensive benefits package to support our employees' well-being and professional growth:

  • 100% covered health benefits (medical, vision, and dental).

  • 401(k) plan with a generous 4% company match.

  • Unlimited paid time off (PTO) policy.

  • Annual $2,000 wellness stipend.

  • Annual $1,000 learning and development stipend.

  • Daily lunches and snacks are provided in our office!

  • Relocation assistance for employees moving to the Bay Area.

Top Skills

Kubernetes
Linux
Pulumi
Python
Terraform
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: Redwood City, California
24 Employees
On-site Workplace
Year Founded: 2023

What We Do

DatologyAI builds tools to automatically select the best data on which to train deep learning models. Our tools leverage cutting-edge research—much of which we perform ourselves—to identify redundant, noisy, or otherwise harmful data points. The algorithms that power our tools are modality-agnostic—they’re not limited to text or images—and don’t require labels, making them ideal for realizing the next generation of large deep learning models. Our products allow customers in nearly any vertical to train better models for cheaper.

Similar Jobs

Snap Inc. Logo Snap Inc.

Staff Software Engineer, ML Infrastructure, 9+ Years of Experience

Artificial Intelligence • Cloud • Machine Learning • Mobile • Software • Virtual Reality • App development
Hybrid
4 Locations
5000 Employees
195K-343K Annually

Snap Inc. Logo Snap Inc.

Software Engineer, ML Infrastructure, 2+ Years of Experience

Artificial Intelligence • Cloud • Machine Learning • Mobile • Software • Virtual Reality • App development
Hybrid
5 Locations
5000 Employees
133K-235K Annually

Snap Inc. Logo Snap Inc.

Software Engineer, ML Infrastructure, 6+ Years of Experience

Artificial Intelligence • Cloud • Machine Learning • Mobile • Software • Virtual Reality • App development
Hybrid
5 Locations
5000 Employees
178K-313K Annually

Snap Inc. Logo Snap Inc.

Principal Software Engineer, Machine Learning Infrastructure

Artificial Intelligence • Cloud • Machine Learning • Mobile • Software • Virtual Reality • App development
Hybrid
5 Locations
5000 Employees
235K-414K Annually

Similar Companies Hiring

True Anomaly Thumbnail
Software • Machine Learning • Hardware • Defense • Artificial Intelligence • Aerospace
Colorado Springs, CO
131 Employees
Caliola Engineering Thumbnail
Software • Machine Learning • Hardware • Defense • Data Privacy • App development • Aerospace
Colorado Springs, CO
53 Employees
Red 6 Thumbnail
Virtual Reality • Software • Hardware • Defense • Aerospace
Orlando, Florida
113 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account