aijobs.net

Senior ML Framework Performance Engineer - AI for Science at Scale

US, CA, Santa Clara, United States
Apply

USD 184K-287K Senior-level Full Time Found 12d ago

Tasks
Perks/Benefits
Skills/Tech-stack

AI for Science | Atomistic modeling | CUDA | Deep understanding | Deep understanding of compute bottlenecks | Distributed Learning | JAX | ML frameworks | Model Parallelism | Parallel Computing | Performance Engineering | PyTorch | Software Design | Triton | WARP

Education

BSc in Computer Science | Computational Biophysics | Computational Chemistry | Doctorate | Master's | Mathematics | Physics

Roles

Engineer | Machine Learning Performance Engineer | Performance Engineer | Senior Machine Learning Performance Engineer

Regions

North America

Countries

United States

States

California, US

Cities

Santa Clara, California, US

Apply
Language: en | Views: 1 | Clicks: 0

Related jobs