4 days ago
Foster City, CA, USAMid Level / Senior
H1B Sponsor
Base Salary
$150k - $350k/yr
Responsibilities
- Work with high-dimensional scientific data formats and contribute to backend compatibility.
- Define engineering patterns for team and collaborating researchers.
- Solve infrastructure problems that determine scientific possibilities.
- Deploy models to production and manage artifact tracking.
- Design and optimize GPU-native data loading pipelines.
- Simplify and improve codebase abstractions.
- Build and maintain pre-training infrastructure for large-scale training runs.
- Cultivate best practices in MLOps across the ML lifecycle.
- Execute complex modifications to the research pipeline.
- Handle DevOps responsibilities to enhance productivity.
- Collaborate with partner researchers to deploy technology.
Requirements
- Hands-on experience with PyTorch, including custom training loops.
- Familiarity with GPU-native data I/O tools and large-scale tensor formats.
- Experience with distributed computing frameworks like Apache Spark or Dask.
- Familiarity with containerization and orchestration tools such as Docker and Kubernetes.
- Experience building or working with AI agent frameworks is a plus.
- A track record of building scalable systems for engineers and researchers.
Benefits
- Generous employer match on employee 401(k) contributions.
- Paid time off to volunteer at an organization of your choice.
- Funding for select family-forming benefits.
- Relocation support for employees who need assistance moving.