aijobs.net

Senior GenAI Research Engineer - Optimization and Kernels

Mountain View, California; San Francisco, California

USD 166K-225K Senior-level Full Time

Apply Save
Found 5d ago
Tasks
Perks/Benefits
Skills/Tech-stack

Activation checkpointing | Attention Mechanisms | CUDA | Collective operations | Data parallelism | DeepSpeed | Distributed Training | FSDP | Gradient Sharding | Gradient synchronization | Hyperparameter Tuning | Kernel Fusion | Megatron-LM | Mixed Precision | Mixed-precision training | NVIDIA GPU | NVProf | Nsight | Pipeline parallelism | PyTorch | PyTorch DDP | Python | Tensor Parallelism | Tensor cores | Zero

Education

Bachelor of Science | Master of Science | PhD

Roles

Engineer | Learning Engineer | Machine Learning Engineer | Research Engineer

Regions

North America

Countries

United States

States

California, US

Cities

San Francisco, California, US | Mountain View, California, US

Apply Save
Language: en Views: 2 Clicks: 0 Saves: 0

Related jobs