Senior Software Engineer, Personalization
Lead personalization and classification algorithm productionization, building ML model-serving infrastructure and translating experimental outputs into scalable production systems. Requires 5+ years experience, backend/infrastructure specialization, and proficiency in AWS, Java, Python, and TypeScript.
Key Responsibilities
- Lead the productionalization of personalization and classification algorithms
- Build with data and analytics in mind to ensure continued accuracy and relevance of personalization, classification, and ranking algorithms
- Translate experimental outputs (from AWS Sagemaker/Personalize) into production-ready systems, ensuring efficiency and reproducibility
- Demonstrate full stack experience with specialization in backend or infrastructure while contributing to broader projects
- Collaborate on architecture and technical decisions that influence the direction of the platform, ensuring scalability, performance, and user-focused improvements
- Design and maintain model-serving infrastructure, including APIs, batch pipelines, or streaming systems required to deploy ML models reliably
- Focus on performance optimization and system reliability, especially as user base grows
- Drive experimentation with MVPs, balancing rapid iteration with long-term sustainable growth and developer experience
- Provide mentorship to engineers, fostering a culture of growth and collaboration
Requirements
- 5+ years of experience leading complex technical projects with significant business impact
- Experience integrating LLM or ML outputs into core product features
- Strong specialization in backend or infrastructure systems while maintaining broad knowledge of the full stack
- Proficiency in IaC, AWS, Java, Python, Typescript with experience in backend systems
- Experience with scalability, performance optimization, and building tools that support data-driven decision-making
- Strong problem-solving skills and a passion for creating products that connect people in the real world
- Ability to lead initiatives, mentor others, and communicate effectively across technical and non-technical teams
Staff Software Engineer, AI Runtime
Staff Software Engineer building and scaling Databricks' managed large-scale GPU training platform (AIR). Focus on distributed training performance, scheduling, fault tolerance, and developer experience for thousands of accelerators.
Senior Software Engineer, AI Runtime
Senior Software Engineer building and scaling Databricks' managed GPU training platform (AI Runtime) for large-scale distributed AI model training. Requires 5+ years in distributed systems and hands-on experience with GPU training frameworks.
Sr. Machine Learning Engineer, Computer Vision
Build and prototype diffusion-based text-to-image generative models (Pinterest Canvas) using large-scale visual-text datasets. Requires 5+ years industry computer vision experience and an M.S. or Ph.D.
Senior AI/ML Engineer
Senior AI/ML Engineer building transformer and deep learning models on financial and behavioral data to power personalized growth and marketing experiences at Chime. Requires strong production ML experience with PyTorch, AWS, and large-scale data infrastructure.