AI Engineer, Intern

📋 Description

• Partner with product teams to translate requirements into AI problems.
• Prepare data pipelines and AI store design under senior guidance.
• Build/validate AI models with PyTorch or JAX (Optax/Orbax/TensorStore/Grain).
• Run large-scale experiments, evaluate models against metrics.
• Support ablation studies and error analysis.
• Help build inference APIs and batch scoring workflows.

🎯 Requirements

• Pursuing BS/MS/PhD in CS, DS, or related field.
• Hands-on ML systems experience via projects, research, or internships.
• Solid Python/Rust fundamentals: data structures, functions, testing, code reviews.
• Exposure to PyTorch or JAX; Pandas and SQL.
• Nice to have: cloud (AWS/GCP/Azure), Docker, or model-serving tools.
• Strong written and verbal communication for cross team docs.

🎁 Benefits

• Full medical coverage.
• Flexible PTO.
• Wellness reimbursement and health programs.
• Monthly lunch stipend.
• Wellness programs to stay healthy in body and mind.
• Team building events and donation matching program.