Member of Technical Staff - GPU Performance Engineer

Remote Remote · San Francisco

pytorch research & engineeringfulltime

ashby

Match & cover letter

Create a profile to see your match and get a cover letter for this role.

Create profile and match

Job description

You will design and implement high-performance GPU kernels for AI systems in a remote role. Collaborate with researchers to optimize workflows and ensure efficient performance across various applications.

Details

Remote work available
Competitive salary with equity
Experience with custom CUDA kernels required
Proficiency in C/C++ and PyTorch

The work

Write high-performance GPU kernels for novel model architectures
Integrate kernels into PyTorch pipelines
Profile and optimize training and inference workflows
Build and maintain performance benchmarks