Staff Machine Learning Engineer
Job Description
About the Role
Our Machine Learning Engineering team powers personalized experiences for hundreds of millions of customers across thousands of brands. We build advanced ML models that predict customer behaviors in real-time, enabling highly personalized shopping experiences. Joining our team offers a high-growth career opportunity to work with some of the world’s most talented machine learning engineers in a high-performance and high-impact culture.
We are seeking a self-driven and highly motivated Machine Learning Engineer to join our growing machine learning teams. As an early hire, you will contribute to the development of machine learning models and infrastructure needs across the Attentive platform and work with Product Management and Engineering to implement end-to-end modeling use cases.
What You'll Accomplish
You have a proven track record of building systems that maintain a high bar of qualityYou deeply loathe regressions and take proactive steps to protect against them through a variety of testing techniquesYou are a collaborator, technical leader, and a great communicatorYou are constantly improving the quality of the project you are working on, both via direct contributions as well as long-term advocacy for larger-scale changesYou are enthusiastic about the high impact, fast-paced work environment of an late-stage startup10+ years experience is idealYour Expertise
You have worked professionally building systems for 6+ years with experience on a single system long enough to see the consequences of your decisionsExperience with TensorFlow/Pytorch, xgboost, pandas, matplotlib, SQL, Spark or similar toolsYou have proficiency or experience with PythonYou have extensive experience using machine learning and data analysis, or similar, to build scalable systems and data-driven products, working with cross-functional teamsYou have a proven track record of building scalable, efficient, automated processes for large-scale data analyses, model development, model validation, and model implementation from modern researchYou have led cross-functional machine learning projects across teamsWhat We Use
Our infrastructure runs primarily in Kubernetes hosted in AWS’s EKSInfrastructure tooling includes Istio, Datadog, Terraform, CloudFlare, and HelmOur backend is Java / Spring Boot microservices, built with Gradle, coupled with things like DynamoDB, Kinesis, AirFlow, Postgres, Planetscale, and Redis, hosted via AWSOur frontend is built with React and TypeScript, and uses best practices like GraphQL, Storybook, Radix UI, Vite, esbuild, and PlaywrightOur automation is driven by custom and open source machine learning models, lots of data and built with Python, Metaflow, HuggingFace 🤗, PyTorch, TensorFlow, and Pandas