aijobs.net

Research MLE (Training Optimization)大模型训练优化工程师

Beijing, Beijing, China

CNY 144K-240K (estimate) Senior-level Full Time

Apply Save
Found 6h ago
Tasks
Perks/Benefits
Skills/Tech-stack

C++ | CUDA | DeepSpeed | Distributed Training | FSDP | GPU Kernel | GPU kernel optimization | Gradient Checkpointing | JAX | Kernel optimization | Low Precision | Megatron-LM | NVIDIA Nemo | PyTorch | Python | Rust | Triton | Zero

Education

N/A

Roles

Engineer | Learning Engineer | Machine Learning Engineer | Research Scientist | Scientist

Regions

Asia/Pacific

Countries

China

States

Beijing, CN

Cities

Beijing, Beijing, CN

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

Related jobs