Machine Learning Infra Engineer

DS creates systems that power the next generation of radio spectrum intelligence. We collect radio data from all over the world, train neural networks to decipher it, and run them on the smallest chips we can. We’re solving a new, technically hard problem where nothing from other fields works out of the box, and along the way, we’ve built our own stack from scratch, including entirely new embedding model architectures, custom GPU kernels, and much more.

Requirements

• Experience designing and implementing data infrastructure from scratch, including databases, cloud storage, and cloud compute
• Experience managing a production-grade Python codebase that was used by other people
• Experience with AWS, including AWS networking, S3, Sagemaker, RDS, ECS, Lambda, and related infra-as-code tools
• Experience designing database schemas, metadata states, and software abstractions that promote clarity and generalize well to new situations
• Experience working directly with researchers and using infrastructure that supports experiment tracking, model versioning, and artifact deployment, such as MLflow or similar
• Experience with larger-than-memory data inexpensively, without setting up a cluster
• Ability to write clearly
• Collaborative attitude and interest in helping define large areas of our engineering roadmap

Benefits

• Above-market salary
• Equity
• Benefits package
• 401(k) match - up to 4% of your salary
• Flexible PTO
• Daily office lunches