POLYROOK
← Back to Jobs
Research Engineer — Foundation Models

Research · Palo Alto, CA · Full-time · Hybrid

$220k – $300k + Equity

Join a small team pushing the limits of scalable vision-language and 3-D priors.

Responsibilities

  • Implement distributed training on 1k+ H100 cluster (DeepSpeed, FSDP).
  • Curate synthetic & human-rated datasets for scene understanding.
  • Benchmark models on academic + internal creator metrics; publish findings.

Minimum Qualifications

  • Strong PyTorch or JAX fundamentals; contributed to large-scale ML infra.
  • Familiarity with NeRF, point-cloud or mesh representation learning.
  • MS/PhD in ML, CV, or equivalent track record (top-tier papers a plus).

Personal Information

Links

Experience

Résumé / CV

Cover Letter (Optional)

Logistics