6 дней назад
Engineer (AI/ML Systems)
Мэтч & Сопровод
Для мэтча с этой вакансией нужен Plus
Описание вакансии
Текст:
TL;DR
Engineer (AI/ML Systems) (AI/ML performance): Building analytical models, benchmarks, and prototypes for large-scale machine learning training and inference on Cerebras hardware and GPU or software baselines with an accent on performance trade-offs across latency, throughput, memory, communication, and compute. Focus on characterizing emerging algorithms, identifying kernel and system bottlenecks, and guiding hardware–software co-design through quantitative analysis.
Location: Sunnyvale, California, United States; hybrid work arrangement.
Company
builds large-scale AI hardware and software systems designed to accelerate machine learning training and inference beyond conventional GPU architectures.
What you will do
- Build analytical and empirical performance models for advanced ML training and inference algorithms.
- Analyze how model size, sequence length, batch size, parallelism, and hardware scale affect algorithmic trade-offs.
- Construct Pareto frontiers covering model quality, latency, throughput, memory, communication, and compute cost.
- Develop prototypes and benchmarks for the WSE and relevant GPU or software baselines.
- Identify kernel, compiler, runtime, communication, and algorithmic bottlenecks through profiling and system analysis.
- Partner with research, kernel, compiler, runtime, inference, and architecture teams on implementation and hardware–software co-design decisions.
Requirements
- Degree or equivalent practical experience in computer science, computer engineering, electrical engineering, mathematics, or a related field.
- Strong knowledge of computer architecture, parallel computing, systems performance, machine learning fundamentals, and ML systems.
- Experience with analytical performance modeling, algorithmic complexity analysis, benchmarking, or system simulation.
- Strong first-principles reasoning about compute, memory, communication, accuracy, and scaling behavior.
- Proficiency in Python and comfort with C++.
- Experience profiling and debugging performance in ML, HPC, CPU, GPU, or accelerator-based systems.
Nice to have
- Experience with roofline analysis, CPU or GPU simulators, kernel optimization, or hardware–software co-design.
- Familiarity with CUDA, Triton, PyTorch, JAX, or open-source LLM training and inference systems.
- Knowledge of transformer internals, attention variants, KV-cache strategies, model parallelism, sparsity, quantization, or parallel generation.
- Research publications, patents, or significant open-source contributions in ML systems, computer architecture, or computational efficiency.
- Experience evaluating technology choices for future hardware or software architectures.
Culture & Benefits
- Work on an AI platform designed to go beyond GPU constraints.
- Opportunities to publish and open-source AI research.
- Work with a high-performance AI supercomputer.
- Combination of startup vitality and job stability.
- Simple, non-corporate culture focused on individual beliefs, learning, growth, and inclusion.
Будьте осторожны: если работодатель просит войти в их систему, используя iCloud/Google, прислать код/пароль, запустить код/ПО, не делайте этого - это мошенники. Обязательно жмите "Пожаловаться" или пишите в поддержку. Подробнее в гайде →
Похожие вакансии
Baseten
8 дней назад
Software Engineer (AI)
180 000 - 360 000$
8 дней назад
AI Systems Researcher/Engineer (ML)
12 дней назад
Senior Engineering Manager (AI/ML Inference)
250 000 - 300 000$
Anthropic
8 дней назад
Research Engineer, Machine Learning (AI)
500 000 - 850 000$
11 дней назад
Inference Performance Engineer (AI)
10 дней назад