4 часа назад
Member of Technical Staff (Research Infrastructure)
200 000 - 400 000$
Мэтч & Сопровод
Для мэтча с этой вакансией нужен Plus
Описание вакансии
Текст:
TL;DR
Member of Technical Staff (Research Infrastructure) (AI/ML Infrastructure): Building the platform for data ingestion, distributed training, evaluation, and production serving of population-scale human behavior simulations with an accent on GPU utilization, data architecture, and inference efficiency. Focus on optimizing multi-agent serving, scaling distributed training pipelines, managing multi-node GPU clusters, and engineering rigorous scientific evaluations.
Location: San Francisco, United States; on-site
Salary: $200,000–$400,000 USD per year, plus equity and comprehensive benefits.
Company
builds AI infrastructure for simulating human behavior at scale and helping organizations make decisions using verifiable predictions.
What you will do
- Build and operate the ML platform covering data exploration, feature generation, experiment tracking, training orchestration, evaluation, and deployment.
- Optimize data and training pipelines by profiling FLOPs utilization, tokenization, GPU memory, ingestion throughput, and system observability.
- Improve population-scale inference through batching, scheduling, KV-cache reuse, quantization, and other serving optimizations.
- Redesign simulation data architecture, including schemas and ingestion logic for high-variety behavioral data.
- Own the multi-node GPU cluster, including NCCL, RDMA, storage, scheduling, autoscaling, portability, and alerting.
- Build rigorous statistical evaluation tooling and translate research advances in simulation, training, and inference into production improvements.
Requirements
- Deep systems and ML proficiency, including high proficiency in Python and hands-on experience with PyTorch or JAX.
- Experience optimizing modern ML architectures and working with NVIDIA GPUs and technologies such as NCCL, InfiniBand, NVLink, CUDA, or Triton.
- Experience building production ML platforms or MLOps systems for training orchestration, experiment tooling, model serving, or LLM applications.
- Experience architecting, observing, debugging, and scaling production distributed systems, ideally performance-critical systems.
- Understanding of the training, fine-tuning, continuous ingestion, monitoring, and high-availability serving lifecycle.
- Strong quantitative foundation and research/data literacy; a background in computer science, mathematics, statistics, or a related field is typical.
Nice to have
- Experience optimizing inference for multi-agent or agentic environments with interdependent requests.
- Experience with distributed training at scale.
- Experience shipping production AI agents and familiarity with vLLM, SGLang, TensorRT-LLM, or inference-time optimization.
- Interdisciplinary experience in social science modeling or behavioral economics.
- Experience writing custom GPU kernels.
Culture & Benefits
- Competitive compensation combining base salary, equity, and comprehensive benefits.
- Medical, dental, and vision coverage.
- Flexible time off supporting work-life balance.
- Emphasis on thoughtful conversations, clear examples of past work, humility, collaboration, and pragmatic ownership.
Hiring process
- The hiring journey uses thoughtful conversations and clear examples of past work to assess mutual fit, working style, and expectations.
- Candidates may reapply for the same role after a 90-day waiting period.
Будьте осторожны: если работодатель просит войти в их систему, используя iCloud/Google, прислать код/пароль, запустить код/ПО, не делайте этого - это мошенники. Обязательно жмите "Пожаловаться" или пишите в поддержку. Подробнее в гайде →
Похожие вакансии
3 часа назад
Member of Technical Staff, Kernels (AI)
200 000 - 350 000$
3 часа назад
Research Scientist (Reinforcement Learning)
200 000 - 335 000$
6 часов назад
ML Software Engineer (Physical AI)
100 000 - 300 000$
3 часа назад
Member of Technical Staff (Applied AI)
150 000 - 350 000$
2 часа назад
Member of Technical Staff, Infrastructure & Training Systems (AI)
3 часа назад
Member of Technical Staff, Training Infra (AI)
200 000 - 350 000$