Senior Principal AI Engineer
Remote - USA, United States
R
USD 129K-190K (estimate) Senior-level Full Time
Tasks
- Apply memory optimization techniques
- Balance compute memory and communication
- Design distributed training systems
- Diagnose and resolve compute memory and network bottlenecks
- Ensure efficient scheduling isolation and fairness
- Improve training stability and fault tolerance
- Minimize networking bottlenecks
- Optimize GPU cluster orchestration
- Optimize and debug distributed communication
- Optimize multi node multi GPU execution
- Own multi node launch configurations and failure recovery
- Scale large model training with distributed frameworks
Perks/Benefits
- N/A
Skills/Tech-stack
Activation checkpointing | Cluster Orchestration | Data parallelism | DeepSpeed | Distributed communication | Fault Tolerance | GPU Cluster | GPU cluster orchestration | Infiniband | Kubernetes | Megatron-LM | Memory Optimization | NCCL | NVLink | Pipeline parallelism | PyTorch distributed | RDMA | Ray | RunAI | Slurm | Tensor Parallelism | Zero
Education
N/A
Related jobs
-
Senior Principal Software Engineer, Machine Learning USD 222K-453KA/B | A/B Testing | B testing | Canary Deployment | Capacity PlanningHybrid work model | MentorshipSenior-level Full TimeRemote, US R2d ago
-
Staff ML Engineer, Perception Research USD 251K-310K3D Perception | Autoregressive models | Data Pipelines | Data parallelism | Diffusion Models401k match | Baby bonding leave | Dental insurance | Disability insurance | Health insuranceSenior-level Full TimeMountain View, CA, USA ; New … R2d ago
-
Senior Software Engineer, Training & Experimentation USD 180K-250KAPI Development | AWS | Asynchronous processing | Automated testing | Azure401k matching | Company holidays | Company winter break | Dental coverage | Flexible scheduleSenior-level Full TimeLondon, England, United Kingdom; New York, … R2d ago
-
Software Engineer, Machine Learning Platform USD 200K-330KAlerting | CI/CD | Cloud Computing | Distributed Systems | Docker401k match | Health, dental, vision insurance | Paid time off | Professional development | Work-life balanceMid-level Full TimeEmeryville, California, United States; Hybrid (2-3 … R2d ago
-
Software Engineer II, Model Platform USD 149K-214KAWS | Data Pipelines | Django | Golang | KubernetesMid-level Full TimeRemote - USA R2d ago
-
AI Search | AI Security | AWS | AWS Bedrock | Agent memoryHybrid work | In-person interviewSenior-level Contract Full TimeJersey City, NJ, United States R2d ago
-
Lead AI Software Engineer, AI Platform & Architecture USD 180K-225KBackend Development | CI/CD | Cloud Architecture | Cost Optimization | Distributed SystemsDental insurance | Flexible time off | Medical insurance | Paid parental leave | Sabbatical after 5 yearsSenior-level Full TimeUnited States (Remote-First) R2d ago
-
Principal AI Engineer – Agentic AI / Generative AI | Irvine, CA (Hybrid) | Contract to Hire | AS A USD 100K-150KAI Agents | API | AWS | Agent systems | CI/CDContract-to-hire | Hybrid workSenior-level Full TimeIrvine, CA R2d ago
-
ML Engineer, II - Simulation Enablement USD 153K-183KAWS | CI/CD | Cloud Storage | Cloud compute | Daft100 percent medical dental and vision premiums | 401k employer match | AD and D insurance | Company holiday office closures | Life insuranceMid-level Full TimeRemote - US, Ann Arbor, MI R2d ago
-
Data Analytics Engineer USD 150K-225KArgo | Argo Workflows | CI/CD | Data Warehousing | Database Design401k matching | Dental insurance | Fully stocked kitchens | Happy hours | In office 4 days per weekMid-level Full TimeNew York, NY R2d ago
-
Lead Machine Learning Engineer USD 144K-248KAWS | Agentic AI | Amazon Athena | CI/CD | Docker401k matching | Bereavement leave | Dental insurance | Fitness and wellness stipend | Jury duty leaveSenior-level Full TimeNew York R2d ago
-
Senior, Data Analyst/BI Engineer USD 108K-145KAWS | Airflow | Amazon Redshift | BigQuery | DataStage401k matching | Healthcare dental vision coverage | PTO | Paid Company Holidays | Paid parental leaveSenior-level Full TimeUnited States R2d ago
-
AI / ML Engineer USD 95K-130KAWS | AWS Lambda | AWS Step Functions | Agent Orchestration | Amazon Bedrock401k | Dental insurance | Employee assistance program | Employee stock purchase plan | FSASenior-level Full TimeLincoln - Nelnet Center, United States R3d ago
-
Senior Applied AI Engineer USD 116K-231KAPI Development | AWS | Asynchronous processing | CI/CD | CloudWatchEmployee assistance program | Flexible work options | Health benefits | Wellness programsSenior-level Full TimePA - Philadelphia, 1701 John F … R3d ago
-
AWS | Agent Harnessing | Agentic Workflows | Amazon SageMaker | Apache AirflowEmployee networks | Flexible remote work | Paid adoption leave | Paid parental leave | Paid time offSenior-level Full TimeUSA CA Remote, United States R3d ago
-
AWS | Alerting | Capacity Planning | Cassandra | Chaos EngineeringFlexible work autonomy | Hybrid work schedule | Open source community involvementSenior-level Full TimeNew York City, United States R3d ago
-
Big Data Application Developer USD 100K-150KAWS EMR | Airflow | Apache Atlas | Apache Flink | Apache HudiRemote workSenior-level Full TimeUnited States - Remote R3d ago
-
Senior-level Full TimeUnited States - Remote R3d ago
-
ML Systems Engineer USD 145K-165KAutoscaling | C++ | Capacity Planning | Distributed Systems | GPU ArchitectureRemote workSenior-level Full TimeUnited States - Remote R3d ago
-
Foundation Model Engineer USD 200K-230KCluster operations | DPO | Distributed Training | Evaluation methodology | FSDPSenior-level Full TimeUnited States - Remote R3d ago
-
ZoomAI DevOps Engineer USD 87K-186KAWS | AWS Bedrock | Ansible | Ansible Tower | ArgoCDHybrid work option | In-office work option | Remote work optionMid-level Full TimeSan Jose (CA), United States R3d ago
-
Senior AI Engineer – AI Services Platform USD 177K-387KAWS | Agent Orchestration | Azure | Benchmarking | C++Career growth | Health benefits | Hybrid work | In-person work | Mental health supportSenior-level Full TimeSeattle (WA), United States R3d ago
-
Senior Principal Data Engineer USD 150K-195KABAC | AI coding | AI coding assistant | AWS | Azure401k | Dental insurance | Employee stock purchase plan | Equity awards | FSASenior-level Full TimeRemote (US), United States R3d ago
-
Senior Specialty AI Engineer USD 143K-224KAPI Security | BM25 | CI/CD | Cloud Build | DockerHybrid work scheduleSenior-level Full Time142019-NC-300 South Brevard, Charlotte, United States R3d ago
-
Sr. Software Engineer, Customer Experience Engineering USD 200K-287KAPI Design | Cloud Computing | Code review | Container Services | DebuggingSenior-level Full TimeUS-CA-Menlo Park R3d ago