about 4 hours ago
Berkeley, CA, USAIntern
H1B Sponsor
Responsibilities
- Translate product requirements into scoped AI problem statements.
- Prepare data pipeline and AI store design under senior engineer guidance.
- Build and validate AI models using PyTorch or JAX.
- Run large-scale experiments and evaluate models against defined metrics.
- Help build inference APIs and batch scoring workflows.
- Support AIOps practices including model versioning and CI/CD pipelines.
- Contribute to root-cause analysis on production issues.
- Document experiments, design decisions, and runbooks.
Requirements
- Currently pursuing a BS/MS/PhD in Computer Science, Data Science, or a related field.
- Some hands-on experience with ML systems through projects or internships.
- Solid Python/Rust fundamentals including data structures and basic testing.
- Exposure to AI/ML frameworks like PyTorch or JAX and basic data tooling.
- Familiarity with cloud platforms, Docker, or model-serving tools is a plus.
- Clear written and verbal communication skills.
Benefits
- Comprehensive medical coverage.
- Flexible PTO and wellness reimbursement.
- Monthly lunch stipend.
- Wellness programs for physical and mental health.
- Frequent team-building events.
- Donation-matching program for causes you care about.