Staff Machine Learning Engineer

People In AI is a high-growth, privacy-first AI startup transforming how enterprises convert third-party data into high-accuracy, ML-ready features. As a Staff Machine Learning Engineer, you will build and ship production ML systems focusing on real-time inference and scalability while impacting business outcomes.

Responsibilities:

• Build and deploy large-scale representation learning and personalization models
• Develop embeddings, ranking systems, recommendation engines, and CTR models
• Design real-time and low-latency inference pipelines
• Own the full ML lifecycle from data processing to production monitoring
• Improve model performance and manage drift in live systems

Requirements:

• Experience shipping ML models into production at scale
• Strong background in embeddings, representation learning, CTR, or sequence modeling
• Hands-on experience with real-time or near-real-time ML systems
• Solid understanding of data pipelines, orchestration, and experimentation tooling
• Product mindset with strong ownership and startup experience
• Python
• Spark
• Airflow
• CI/CD pipelines
• PyTorch or TensorFlow
• Distributed data and inference systems