PyTorch Jobs in Beijing, China

AI Engineer

Machine Learning Infrastructure

2w ago

PO

Member of Engineering (Scalability)

Poolside

GB or Remote (Worldwide)

R&D

Linux Kernel

Machine Learning Engineering

3w ago

BA

Software Engineer - Model Performance

Baseten

San Francisco, California, US or Remote (Worldwide)$180K – $360K Yearly

Software Engineering

C++

3w ago

CO

Member of Technical Staff, Training Performance Engineer

Cohere

London, England, GB or Remote (Worldwide)

Performance Engineer

Machine Learning Engineering

2w ago

SA

Generative AI Inference Engineer

Stability AI

Worldwide (Remote)

Generative AI Engineering

4w ago

NV

Senior Software Engineer, JAX

NVIDIA

Worldwide (Remote)

Senior Software Engineer

JAX Developer

C/C++

2d ago

NV

Deep Learning Solution Architect

NVIDIA

Beijing, Beijing, China (On-site)

Solution Architecture

Deep Learning

Reinforcement Learning

Large Language Models

3w ago

CO

Member of Technical Staff, Training Infra Engineer

Cohere

Paris, FR or Remote (Worldwide)

Infrastructure Engineering

AI/ML Software Development

3w ago

NV

AI Computing Software Development Engineer, TensorRT

NVIDIA

Beijing, Beijing, China (On-site)

Deep Learning Engineering

TensorRT

C++ Programming

2w ago

SA

Multimodal Generative AI Researcher

Stability AI

Worldwide (Remote)

Multimodal AI Research

Vision-Language Model Research

DeepSpeed

3w ago

CO

Senior ML Systems Engineer, Frameworks & Tooling

Cohere

London, England, GB or Remote (Worldwide)

MLOps

Distributed Systems

3w ago

CO

Senior Member of Technical Staff, Multimodal AI

Cohere

San Francisco, California, US or Remote (Worldwide)

AI Research Engineer

Machine Learning Infrastructure

3w ago

BA

Senior Software Engineer - Model Training

Baseten

San Francisco, California, US or Remote (Worldwide)$165K – $330K Yearly

3w ago

BA

Senior Product Engineer - Training Platform

Baseten

San Francisco, California, US or Remote (Worldwide)$165K – $330K Yearly

Distributed Systems Engineering

Kubernetes

Machine Learning Engineering

2d ago

RE

Member of Technical Staff (Machine Learning Engineer)

Reka

Worldwide (Remote)

AI Engineering

Java

2d ago

NV

Solutions Architect - CPU and LPU

NVIDIA

Beijing, Beijing, China (On-site)

Solutions Architecture

AI Infrastructure

NVIDIA Grace

NVIDIA Vera

1w ago

NV

Solution Architecture Intern, AI in Industry - 2026

NVIDIA

Beijing, Beijing, China (On-site)

Solution Architecture

AI Engineering

Accelerated Computing

Linux

3w ago

CO

Staff Research Engineer, Model Efficiency

Cohere

New York, US or Remote (Worldwide)

Modeling

Machine Learning

Model Architecture

3w ago

RE

Member of Technical Staff (ML)

Reka

Worldwide (Remote)