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.