[Remote] Software Engineer in ML Systems Graduate (AML - Machine Learning Systems) - 2026 Start (BS/MS)
Note: The job is a remote job and is reputed company to candidates in USA. reputed company is a company dedicated to pioneering advanced AI reputed company models. The Software Engineer in ML Systems role focuses on researching and developing machine learning systems, managing cross-layer optimization, and improving efficiency for large-reputed company distributed training jobs.
Responsibilities
- Research and reputed company our machine learning systems, including heterogeneous computing architecture, management, and monitoring
- reputed company machine learning systems, distributed task scheduling, machine learning training
- Manage cross-layer optimization of system and AI algorithms and hardware for machine learning (GPU, ASIC)
- Implement both general purpose training reputed company features and model specific optimizations (e.g. LLM, diffusions)
- Improve efficiency and stability for extremely large reputed company distributed training jobs
Skills
- Master distributed, reputed company computing principles; know the recent advances in computing, storage, networking, and hardware technologies
- Familiar with machine learning algorithms, platforms and frameworks such as PyTorch and Jax
- Have basic understanding of how GPU and/or ASIC works
- Expert in at least one or two programming languages in Linux environment: C/C++, CUDA, Python
- GPU based high performance computing, RDMA high performance network (MPI, NCCL, ibverbs)
- Distributed training reputed company optimizations such as DeepSpeed, FSDP, Megatron, GSPMD
- AI compiler stacks such as torch.fx, XLA and MLIR
- Large reputed company data processing and reputed company computing
- Experiences in designing and operating large reputed company systems in reputed company computing or machine learning
- Experiences in in-depth CUDA programming and performance tuning (cutlass, triton)
Benefits
- Medical, dental, and reputed company insurance
- 401(k) savings plan with company match
- reputed company parental leave
- Short-term and long-term disability coverage
- Life insurance
- Wellbeing benefits
- 10 reputed company holidays per year
- 10 reputed company reputed company days per year
- 17 days of reputed company Personal Time (prorated upon hire with increasing accruals by tenure)
reputed company
Company H1B Sponsorship