Label Your Data logo

Performance Engineer, GPU

Label Your Data·$280,000 – $850,000

Primary stack

PyTorchJAX

Job description

About the Position

Performance Engineer, GPU

San Francisco, CA | New York City, NY | Seattle, WA


About Anthropic

Anthropic’s mission is to create reliable, interpretable, and steerable AI systems. We want AI to be safe and beneficial for our users and for society as a whole. Our team is a quickly growing group of committed researchers, engineers, policy experts, and business leaders working together to build beneficial AI systems.


About the Role

Pioneering the next generation of AI requires breakthrough innovations in GPU performance and systems engineering. As a GPU Performance Engineer, you'll architect and implement the foundational systems that power our models and push the frontiers of what's possible with large language models. You'll be responsible for maximizing GPU utilization and performance at unprecedented scale, developing cutting-edge optimizations that directly enable new model capabilities and dramatically improve inference efficiency.

Working at the intersection of hardware and software, you'll implement state-of-the-art techniques from custom kernel development to distributed system architectures. Your work will span the entire stack-from low-level tensor core optimizations to orchestrating thousands of GPUs in perfect synchronization.


Responsibilities

  • Architect and implement foundational systems for large language models
  • Maximize GPU utilization and performance at scale
  • Develop cutting-edge optimizations for inference efficiency
  • Implement state-of-the-art techniques in GPU programming and optimization
  • Collaborate with researchers and engineers to shape the future of AI infrastructure

Requirements

  • Deep experience with GPU programming and optimization at scale
  • Impact-driven mindset with a passion for delivering measurable performance breakthroughs
  • Ability to navigate complex systems from hardware interfaces to high-level ML frameworks
  • Collaborative problem-solving and pair programming skills
  • Experience with CUDA, PyTorch, JAX, XLA, NCCL, NVLink, and more

Nice to Have

  • Experience with GPU Kernel Development: CUDA, Triton, CUTLASS, Flash Attention, tensor core optimization
  • Knowledge of ML Compilers & Frameworks: PyTorch/JAX internals, torch.compile, XLA, custom operators
  • Skills in Performance Engineering: Kernel fusion, memory bandwidth optimization, profiling with Nsight
  • Experience with Distributed Systems: NCCL, NVLink, collective communication, model parallelism
  • Familiarity with Low-Precision: INT8/FP8 quantization, mixed-precision techniques
  • Experience with Production Systems: Large-scale training infrastructure, fault tolerance, cluster orchestration

We Offer

  • Competitive annual salary range: $280,000-$850,000 USD
  • Comprehensive benefits package
  • Optional equity donation matching
  • Generous vacation and parental leave
  • Flexible working hours
  • Lovely office space in San Francisco, New York City, or Seattle
  • Visa sponsorship available

About the Company

Anthropic is a public benefit corporation headquartered in San Francisco. We are committed to creating reliable, interpretable, and steerable AI systems. Our team values impact, collaboration, and communication skills. We encourage diverse perspectives and are dedicated to the ethical implications of AI.


Deadline to apply: None. Applications will be reviewed on a rolling basis.

We encourage you to apply even if you do not believe you meet every single qualification. Not all strong candidates will meet every single qualification as listed. Research shows that people who identify as being from underrepresented groups are more prone to experiencing imposter syndrome and doubting the strength of their candidacy, so we urge you not to exclude yourself prematurely and to submit an application if you're interested in this work.

Your safety matters to us. To protect yourself from potential scams, remember that Anthropic recruiters only contact you from @anthropic.com email addresses. In some cases, we may partner with vetted recruiting agencies who will identify themselves as working on behalf of Anthropic. Be cautious of emails from other domains. Legitimate Anthropic recruiters will never ask for money, fees, or banking information before your first day. If you're ever unsure about a communication, don't click any links-visit anthropic.com/careers directly for confirmed position openings.

© Label Your Data. This job description was sourced from the employer's public career page. TheJob is not the employer — we index the posting and route candidates to the source. All content rights and hiring decisions belong to the employer.

Label Your Data provides secure and high quality data annotation services for Computer Vision or NLP applications since 2020. We help Data Scientists and AI Engineers streamline the dataset labeling and focus on model development.About Label Your Data:— Founded in 2020;— 200+ annotation experts in Europe, LATAM and Africa;— 98%+ annotation accuracy benchmark;— 55 supported languages;— PCI/DSS certified;— ISO/IEC 27001:2013 certified;— GDPR, CCPA and HIPAA-compliant;— HQs in North America, EU and Asia.Our experience spans over 20 industries including Automotive, Robotics, Agriculture, E-commerce, Retail, Healthcare, Manufacturing, Fintech and Insurance.Our core services include but are not limited to:— Image & video annotation: 2D boxes, OCR, object/action detection, polygons, key points, semantic segmentation, 3D cuboids;— Text annotation: classification, NER, intent & sentiment analysis, prompts creation, Q&A pairs generation;— Audio annotation: transcription, sentiment recognition;— Sensor data annotation;— Data classification & categorization;— Data entry;— Data collection;— Model validation.

More at Label Your Data

All 1589 roles

Popular searches