Senior AI Infrastructure Engineer - Model Training
Tasks
- Build and optimize distributed training infrastructure
- Design high throughput data loading and streaming systems
- Develop scalable dataset construction pipelines
- Maximize accelerator utilization with mixed precision training
- Partner with ML teams to scale model training runs
- Profile training pipelines to eliminate bottlenecks
Perks/Benefits
- 401k
- Dental and vision plans
- Dependent care FSA
- Dog-friendly office
- FSA
- Flexible PTO
- Free EV charging
- Free catered lunch
- Gym membership benefits
- HSA
- Legal services
- Life insurance
- Long-term disability
- Medical/Dental/Vision insurance
- Mental health support
- Paid Holidays
- Parental leave
- Short-term disability
Skills/Tech-stack
BF16 | C++ | CUDA | Data parallelism | DeepSpeed | Distributed Training | FP8 | Fused kernels | GPU Profiling | Infiniband | MDS | Megatron | Memory Optimization | Mixed Precision | NCCL | NVIDIA Nsight | NVLink | Pipeline parallelism | PyTorch | PyTorch DDP | PyTorch FSDP | PyTorch Profiler | Python | Tensor Parallelism | Triton | WebDataset | Zero
Education
Regions
Countries
States
Related jobs
-
Featured Feat. Principal Knowledge & Data Architect USD 174K-284KAWS Neptune | Canonicalization | Chunking | Cypher | DBTBenefits including health and wellness programs | Health, wellness, and retirement plansSenior-level Full TimeHeadquarters - Chevy Chase, MD R20d ago
-
Senior-level Full TimeBoston, Massachusetts, United States2h ago
-
Lead Applied AI Engineer II USD 124K-255K.NET | AWS | AWS Bedrock | Agent Orchestration | Agentic AISenior-level Full TimeDallas, Texas, United States; Hermitage, Tennessee, …2h ago
-
Staff Software Engineer USD 110K-165KAI Safety | AWS | Agentic Workflows | Airflow | Automated testingHealth insurance | Onsite work presence required | Paid time off | Retirement contributionsSenior-level Full TimeBurlington, Massachusetts, US, 018038h ago
-
Senior-level Full TimeSeattle, WA12h ago
-
Senior-level Full TimeRedmond, WA12h ago
-
Senior Software Engineer, Data Engineering USD 172K-230KApache Airflow | Apache Spark | Flink | Presto | Python401k matching | Catered meals | Company holidays | Disability insurance | Fertility benefitsSenior-level Full TimeMenlo Park, CA12h ago
-
Lead Software Engineer - Java, AI Lead USD 170K-195KAWS | AWS EC2 | AWS Lambda | AWS S3 | Access ManagementBackup childcare | Financial coaching | Health care coverage | Mental health support | On-site health and wellness centersSenior-level Full TimePlano, TX, United States13h ago
-
Staff ML Engineer, Agent Training & Environments USD 250K-280KApache Kafka | CI/CD | Cloud platform | Direct Preference Optimization | Google CloudCareer growth tied to impact | Hybrid work model | Ownership and autonomySenior-level Full TimeSan Francisco Bay Area R14h ago
-
Senior AI Engineer USD 170K-200KAPI Integration | Anomaly Detection | Artificial Intelligence | Caching | Embeddings401k match | Flexible paid time off | Gym access | HSA contribution | Medical, dental, and visionSenior-level Full TimeRemote, USA R14h ago
-
Software Engineer, Business Systems & Analytics USD 111K-146KAPI Integration | AWS | Agile Scrum | Battery Dispatch | CI/CDEntry-level Full TimeScottsdale, AZ15h ago
-
Senior Software Engineer USD 119K-234KAnomaly Detection | Apache Flink | Apache Kafka | Apache Spark | AzureHybrid workSenior-level Full TimeUnited States, Washington, Redmond15h ago
-
Senior Space Embedded Linux Software Engineer USD 169K-216KAnalog electronics | Bare Metal | C# | C++ | CAN401k employer match | Dental insurance | EV charging | Flexible PTO | HSA employer matchSenior-level Full TimeEl Segundo, California, United States15h ago
-
Senior Software Engineer USD 119K-234KApache Flink | Apache Kafka | Apache Spark | Azure | Azure BlobSenior-level Full TimeUnited States, Washington, Redmond15h ago
-
Space Embedded Linux Software Engineer USD 131K-184KAnalog electronics | Bare Metal | C# | C++ | CAN401k employer match | EV charging | Flexible PTO | HSA employer match | Medical/Dental/Vision insuranceEntry-level Full TimeEl Segundo, California, United States15h ago
-
Senior/Staff Software Engineer, Data Platform USD 155K-185KAWS | AWS CloudFormation | Airflow | Amazon S3 | Apache Spark11 paid holidays | 401k plan | Commuter benefits | Dental insurance | Flexible vacationSenior-level Full TimeBoulder, Colorado15h ago
-
AWS Glue | AWS S3 | Amazon Kinesis | Ansible | Apache Flink401k | Dental insurance | Medical insurance | Paid sick hours | Vision insuranceMid-level Contract Full TimeEnglewood Cliffs, NJ, United States15h ago
-
Senior AI/ML Engineer USD 134K-197KAWS | Automated testing | Azure | CI/CD | Containerization401k match | Flexible PTO | Flexible spending account | Health, dental, and vision insurance | Paid Cloud Developer AccountsSenior-level Full TimeArlington, VA15h ago
-
Embedded Engineer - Streaming USD 130K-280KAV1 | Adaptive bitrate | Adaptive bitrate streaming | C# | C++Commuter benefits | Dental insurance | Fertility benefits | Flexible PTO | Flexible Spending Account (FSA)Senior-level Full TimeSan Mateo, CA United States15h ago
-
AI/ML Engineer USD 86K-180KAPI Development | AWS | Automated testing | Azure | CI/CD401k match | Access to sports tickets | Dental insurance | Federal Holidays | Flexible PTOMid-level Full TimeArlington, VA16h ago
-
DataOps Engineer USD 110K-140KAWS Glue | AWS S3 | Amazon Kinesis | Ansible | Apache FlinkOn-call rotation | Relocation requiredMid-level Contract Full TimeEnglewood Cliffs, NJ, United States16h ago
-
Lead Software Engineer - Big Data USD 175K-186KACID | Apache Cassandra | Apache HBase | Apache Iceberg | Apache SparkBackup childcare | Health care coverage | Mental health support | Retirement savings plan | Tuition reimbursementSenior-level Full TimePlano, TX, United States16h ago
-
Analytics Engineer USD 138K-215KAccess Control | Data Documentation | Data Glossary | Data Governance | Data Lineage401k match | Dental insurance | Health insurance | Health savings account | Paid time offSenior-level Full TimeCottonwood Heights, Utah, United States16h ago
-
Space Embedded Software Engineer, C++ USD 131K-184KBare Metal | C++ | Computer Architecture | Computer Networking | Data Structures401k employer match | EV charging | Flexible PTO | Gym and fitness reimbursement | HSA employer matchEntry-level Full TimeEl Segundo, California, United States16h ago
-
Senior Space Embedded Software Engineer, C++ USD 169K-216KBare Metal | C++ | Computer Architecture | Computer Networking | Data Structures401k with employer match | Cell phone reimbursement | Company-provided meals | Dental insurance | EV chargingSenior-level Full TimeEl Segundo, California, United States16h ago