1. Home
  2. Jobs
  3. PyTorch/XLA

PyTorch/XLA jobs

Explore PyTorch/XLA roles on Inference Jobs and apply today.

41-60 of 334 jobs

NV2w

Senior Machine Learning Applications and Compiler Engineer

NVIDIA

Santa Clara, California, United States (Hybrid)

$152k – $287.5k Yearly

TA1w

Machine Learning Engineer - Inference

Together AI

San Francisco, California, United States (On-site)

$160k – $230k Yearly

NV1w

Senior LLM Agents Architect

NVIDIA

Yokneam Ilit, Northern District, Israel (Hybrid)

GR5d

Senior Principal Performance Engineering

Graphcore

Austin, Texas, United States (Hybrid)

PE2w

Internship - Search Machine Learning Engineer (Belgrade)

Perplexity

Belgrade, Belgrade, Serbia (On-site)

MA2w

AI Scientist - Audio

Mistral AI

Île de Ré, Charente-Maritime, France (Hybrid)

CO3w

Software Engineer, Inference AI/ML

CoreWeave

Sunnyvale, California, United States (Hybrid)

$92k – $135k Yearly

PE2w

AI Inference Engineer (San Francisco)

Perplexity

San Francisco, California, United States (On-site)

$210k – $385k Yearly

OP1w

Inference Runtime, Engineering Manager

OpenAI

San Francisco, California, United States (On-site)

$455k – $555k Yearly

NE4d

Senior ML Engineer (Token Factory)

Nebius

Netherlands + 4 more (Remote)

NV3w

GPU Computing Engineer - Autonomous Driving

NVIDIA

Shanghai, Shanghai, China (On-site)

MA1w

AI Scientist - Warsaw

Mistral AI

Warszawa, Masovian Voivodeship, Poland (Hybrid)

PE2w

AI Infra Engineer (San Francisco)

Perplexity

San Francisco, California, United States (On-site)

$210k – $385k Yearly

CO1w

Principal Engineer, Inference

CoreWeave

Sunnyvale, California, United States (Hybrid)

$206k – $303k Yearly

XA1w

Member of Technical Staff, Recommendation Systems

xAI

Palo Alto, California, United States (On-site)

$180k – $440k Yearly

PO2w

Member of Engineering (Scalability)

Poolside

United Kingdom or Remote (Europe, Middle East, and Africa, North America)

NV1w

Senior GPU Functional Modeling Architect

NVIDIA

Santa Clara, California, United States (On-site)

$152k – $287.5k Yearly

CE1w

Senior Research Engineer - Inference ML

Cerebras

Sunnyvale, California, United States (Hybrid)

OP2w

Research-Hardware Codesign Engineer

OpenAI

San Francisco, California, United States (Hybrid)

$230k – $460k Yearly

TE1w

Sr. Software Engineer, AI Compiler

Tenstorrent

Toronto, Ontario, Canada (Hybrid)

$100k – $500k Yearly