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 R17h ago
-
Senior Software Engineer, Unsecured Installments USD 160K-223KA/B | A/B Testing | Accessibility | B testing | CI/CD401k match | Dental insurance | Disability insurance | Health insurance | Health savings accountSenior-level Full TimeUnited States | Remote R5h ago
-
Software Engineer, Unsecured Installments USD 136K-196KCI/CD | Cypress | GraphQL | Java | JavaScript401k | Company holidays | Dental insurance | Employee assistance program | Health insuranceSenior-level Full TimeUnited States | Remote R5h ago
-
Senior Machine Learning Engineer, Voice Assistant USD 195K-361KAmazon EMR | Amazon S3 | Apache Airflow | Apache Spark | Artificial IntelligenceFinancial wellness support | Flexible remote work | Mental health support | Paid time offSenior-level Full TimeAustin, Texas5h ago
-
Senior Machine Learning Engineer, Voice Assistant USD 216K-273KAmazon EMR | Amazon S3 | Apache Airflow | Automated Speech Recognition | C++Disability benefits | Equity awards | Health insurance | Hybrid work flexibility | Life insuranceSenior-level Full TimeBoston, Massachusetts5h ago
-
Senior Machine Learning Engineer, Voice Assistant USD 229K-367KAmazon EMR | Amazon S3 | Apache Airflow | Apache Spark | Artificial IntelligenceDisability benefits | Equity awards | Health insurance | Life insurance | Paid time offSenior-level Full TimeSan Jose, California5h ago
-
Software Engineer, Embedded UI USD 170K-210KAI | API Design | BrightScript | C++ | Cloud DevelopmentDisability benefits | Health insurance | Life insurance | Paid time off | Parental leaveSenior-level Full TimeSan Jose, California5h ago
-
Technical Program Lead, AI Delivery USD 162K-226KData Modeling | Data Pipelines | Data Science | Generative AI | KPI TrackingClient-facing work | Executive visibility | Hybrid work | Mentorship and team leadershipSenior-level Full TimeSan Francisco6h ago
-
Campus AI Research Engineer (Full-Time) USD 300K-300KC++ | CUDA | GPU Programming | HPC | High ThroughputCPT/OPT eligible | Visa sponsorshipEntry-level Full TimeChicago; New York8h ago
-
Campus AI Research Engineer (Intern) USD 300K-300KC# | C++ | CUDA | Data Mining | Deep learningCPT OPT eligibility | International student support | Work visa sponsorshipEntry-level InternshipChicago; New York8h ago
-
Senior Data Engineer USD 157K-218KAWS | AWS Glue | Amazon Athena | Amazon EKS | Amazon EMRAdoption reimbursement | Commuter stipend | Fertility reimbursement | Flexible PTO | In office meals stipendSenior-level Full TimeDenver, Colorado, United States; San Francisco, …9h ago
-
Applied AI Engineer - Federal (TS Required) USD 170K-300KAgentic Workflows | Airflow | Apache Spark | Chroma | CrewAIAnnual travel up to 25 percent | Federal clearance supportMid-level Full TimeUnited States (Remote); Washington, D.C. (Remote) R9h ago
-
Data Engineer (Starlink) USD 125K-175KAI Agents | APIs | CRM | CRM Data Deduplication | CRM data401k retirement plan | Dental insurance | Employee stock purchase plan | Life insurance | Long-term disability insuranceSenior-level Full TimeHawthorne, CA10h ago
-
Senior-level Full TimeBastrop, TX10h ago
-
Sr. Embedded Security Engineer (Starlink) USD 170K-235KApplied cryptography | C++ | Cryptographic Services | Distributed Systems | Golang401k retirement plan | Company shuttles | Company stock long term incentives | Employee stock purchase plan | Flexible work ambiguity learningSenior-level Full TimeRedmond, WA10h ago
-
Sr. Embedded Security Engineer (Starlink) USD 165K-230KApplied cryptography | Authenticated Boot | C++ | Distributed Systems | GoExtended hours | Weekend work as neededSenior-level Full TimeBastrop, TX10h ago
-
Embedded Security Engineer (Starlink) USD 130K-180KC++ | Cryptography | Go | Hardware Security Module | Hardware securityExtended hours | Weekend workSenior-level Full TimeBastrop, TX10h ago
-
Sr. AI Engineer USD 146K-183KAPI Development | AWS | Agent Frameworks | Azure | Cloud platform401k match | Commuter benefits | Dental insurance | Employee discounts | Medical insuranceSenior-level Full TimeUS - Orlando11h ago
-
Sr. AI Engineer USD 146K-199KAWS | Agentic Workflows | Azure | Cloud platform | Generative AI401k match | Commuter benefits | Dependent care support | Employee discounts | Medical, dental, and vision coverageSenior-level Full TimeUS - Charlotte11h ago
-
Agentic AI Engineer – Google ADK/GCP USD 130K-191KAPI Development | Agent memory | Agent systems | CI/CD | Cloud RunFinancial and income protection benefits | Medical/Dental/Vision | Paid time off (PTO) | Tax-advantaged health care accounts | Wellness time offSenior-level Full TimeAustin, TX R11h ago
-
Finance Analytics USD 122K-143KBusiness Intelligence | Data Modeling | Data Visualization | Data forecasting | Looker401k | Dental insurance | Health insurance | Onsite work | Paid HolidaysMid-level Full TimeLos Angeles, CA12h ago
-
Senior Analytics Engineer (Finance) USD 150K-215KAnalytics engineering | Anomaly Detection | CI/CD | Cortex ML | DBTSenior-level Full TimeBoston, MA12h ago
-
Analytics Engineer II USD 125K-175KCI/CD | Code review | Continuous integration | DBT | Data GovernanceMid-level Full TimeBoston, MA12h ago
-
Sr. Analytics Engineer/ Data Scientist USD 205K-249KAWS | Amazon Redshift | Anomaly Detection | Apache Airflow | Business Intelligence401k | Bi-annual offsites | Commuter stipend | Dental insurance | EquitySenior-level Full TimeSan Francisco, CA12h ago
-
Associate Client Data Engineer USD 110K-175KAWS | AWS Lambda | Airflow | Amazon Redshift | Amazon S3Hybrid work schedule | In person office attendance at least two days per week | Mission driven work culture | Opportunity for mentorshipMid-level Full TimeWashington, DC12h ago