Senior Software Engineer, AI Infrastructure

📋 Description

• Design, build, and maintain scalable ML deployment systems.
• Collaborate with ML teams to streamline workflows and optimize models.
• Scale our feature store for fast real-time and batch retrieval.
• Build observability for model performance and data pipelines.
• Optimize cloud CPU/GPU resources for cost-effective training on AWS.

🎯 Requirements

• 4+ years in software engineering, ML infra or data ops.
• Hands-on with model serving, distributed systems, or production ML workflows.
• Familiar with ML infra tools: Ray, Kubeflow, SageMaker, TensorFlow Serving, Triton.
• Proficient in Python, C++, and ML frameworks like TensorFlow or PyTorch.
• Experience with large-scale search, embeddings, and vector databases (Qdrant).
• Collaborative, building reliable platforms with company-wide impact.

🎁 Benefits

• Challenging, high-impact work to grow your career.
• Performance-based pay with equity, bonuses, and 401(k) matching.
• Best-in-class benefits with 100% paid health insurance.
• Lifestyle wallet for wellness, learning, and more.
• Employer-paid life and disability insurance, fertility benefits.
• Generous time off: holidays, PTO, sick time, parental leave.