Software Engineer, Training & Inference Infrastructure

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 an engineer with deep experience building and operating large-scale training and inference systems. You will design, implement, and maintain the infrastructure that powers both our internal ML research workflows and the high-performance inference pipelines that deliver curated data to our customers.

As one of our early hires, you will influence technical direction, partner directly with researchers and product engineers, and take ownership of systems that are central to our company's success.

What You'll Work On

• Architect and maintain training infrastructure that are reliable, scalable, and cost-efficient.
• Build robust model serving infrastructure for low-latency, high-throughput inference across heterogeneous hardware.
• Automate resource orchestration and fault recovery across GPUs, networking, OS, drivers, and cloud environments.
• Partner with researchers to productionize new models and features quickly and safely.
• Optimize training and inference pipelines for performance, reliability, and cost.
• Ensure all infrastructure meets the highest bar for reliability, security, and observability.

About You

Don't meet every single requirement? We still encourage you to apply. If you're excited about our mission and eager to learn, we want to hear from you!

Compensation

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

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