Research Engineer - Machine Learning

Research Engineer - Machine Learning and Systems

Location: New York City Office HQ

Employment Type: Full time

Department: Research

Overview

We are hiring a principal level Research Engineer with deep strength in machine learning or 3D graphics, software engineering, and systems design. You will bridge frontier research with production systems and ship advanced models used in real products. The work spans exploration, rapid prototyping, rigorous experimentation, and dependable production deployment. Expect to push the limits of spatial intelligence and controllable graphics while keeping systems robust, scalable, and cost efficient.

Role

You will partner with research, engineering, and product to design, build, and operate large models and high performance systems. You will set technical standards, mentor others, and raise the bar for research quality, code quality, and reproducibility.

Key responsibilities

1. Research, design, and implement models and systems across vision, generative modeling, simulation, rendering, and 3D perception
2. Build data, training, evaluation, and deployment pipelines with strong observability and reproducibility
3. Translate research insights into reliable production services that meet product and latency requirements
4. Contribute hands on across prototyping, optimization, integration, and scaling
5. Survey new methods and run grounded evaluations to identify what to adopt and when
6. Share expertise through design reviews, mentoring, and documentation

Minimum qualifications

1. PhD in Computer Science, Machine Learning, Computer Graphics, Computer Vision, or related field, or equivalent research track record
2. Seven or more years of experience in applied ML or research engineering including significant time in fast paced or startup settings
3. Strong publication record in top venues such as NeurIPS, ICLR, ICML, CVPR, ECCV, ICCV, SIGGRAPH, or TOG with multiple first author papers or equivalent impactful artifacts
4. Proven experience training and serving large models at scale including multi GPU or multi node training, distributed data loading, mixed precision, and memory optimization
5. Fluency in Python and C++ and experience writing efficient CUDA or Triton kernels
6. Expertise with PyTorch or JAX and modern tooling for experiment tracking, evaluation, and deployment
7. Demonstrated ability to take ideas from paper to production with measurable impact on users or business outcomes
8. Strong systems skills including profiling, performance tuning, reliability engineering, and cost awareness
9. Excellent communication with the ability to work across research and product teams

Preferred qualifications

1. Contributions that are widely used in the community such as open source libraries, datasets, or benchmarks with visible adoption
2. Experience in neural rendering, differentiable rendering, 3D reconstruction, volumetric video, SLAM, geometric deep learning, or simulation
3. Experience operating large training jobs on Kubernetes, Slurm, or Ray across public cloud environments
4. Experience with evaluation and safety for generative or interactive models including red teaming and guardrail design
5. Track record of mentoring teams and setting research and engineering best practices
6. Patents or awards that recognize technical contributions

Nice to have

1. Shipped interactive graphics or 3D systems with strict real time constraints
2. Experience building custom compilers or graph level optimizations such as CUDA graphs, XLA, or graph capture
3. Prior leadership in cross functional initiatives spanning data, infra, and product

How to apply

Please include a CV, links to publications, code, and a brief summary of two projects that best represent your impact. Include details on model scale, data scale, latency or throughput targets, and the concrete results you achieved.

On site in New York City required. Relocation support available.