Research Scientist / Research Engineer
New York, NYSan Francisco, CAAI ResearchRemote4+ YOE
Summary
Conducts research on efficient AI systems focusing on real-time adaptation, model efficiency, and cross-stack optimization. Requires PhD or equivalent, 4-5+ years industry experience, and deep ML expertise including PyTorch/JAX and optimization techniques.
About the role
Responsibilities
- Innovate on product to co-design algorithms that react real-time to product signal and feedback.
- Design new ways of giving feedback that drive better algorithms.
- Collaborate across software, hardware, and algorithmic domains to achieve system-wide efficiency gains.
- Measure real world impact.
Qualifications
- PhD or equivalent research experience in a computer science field.
- 4-5+ years of industry experience.
- Deep expertise in at least one area: model efficiency, real time alignment or algorithmic optimization.
- Systems thinking ability to understand and optimize across the full ML stack.
- Strong programming skills in Python.
- Experience with deep learning frameworks (PyTorch, JAX, TensorFlow).
- Knowledge of model optimization techniques (RLHF, finetuning).
- Experience in an industry lab with computing at scale.
Nice-to-Haves
- Great teammates who make work feel lighter and bold ideas.
- Adaptable.
Skills
PythonPyTorchJAXTensorFlowRLHFFinetuning