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 R16h 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 R4h 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 R4h 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, Texas4h 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, Massachusetts4h 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, California4h 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, California4h 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 Francisco5h 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, …8h 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, CA9h ago
-
Senior-level Full TimeBastrop, TX9h 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, WA9h ago
-
Sr. Embedded Security Engineer (Starlink) USD 165K-230KApplied cryptography | Authenticated Boot | C++ | Distributed Systems | GoExtended hours | Weekend work as neededSenior-level Full TimeBastrop, TX9h ago
-
Embedded Security Engineer (Starlink) USD 130K-180KC++ | Cryptography | Go | Hardware Security Module | Hardware securityExtended hours | Weekend workSenior-level Full TimeBastrop, TX9h 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, CA11h ago
-
Senior Analytics Engineer (Finance) USD 150K-215KAnalytics engineering | Anomaly Detection | CI/CD | Cortex ML | DBTSenior-level Full TimeBoston, MA11h ago
-
Analytics Engineer II USD 125K-175KCI/CD | Code review | Continuous integration | DBT | Data GovernanceMid-level Full TimeBoston, MA11h 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, CA11h 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, DC11h ago
-
Staff Software Engineer - Data Platform USD 166K-232KAirflow | Automated testing | BigQuery | CI/CD | Cloud platformDental insurance | Flexible time off | In-office workspace | LinkedIn Learning access | Medical insuranceSenior-level Full TimeUnited States (Remote) R12h ago
-
Data Engineer USD 77K-90KAPI Integration | Automation | Batch Processing | CI/CD | CSV401k | Dental insurance | Health insurance | Life insurance | Paid time offMid-level Full TimeRemote, USA, United States R12h ago
-
Robotics Software Engineer USD 100K-150KC++ | CI/CD | Camera | Feedback control | Git401k retirement plan | Dental coverage | Employee referral bonuses | Flexible PTO | Free lunchEntry-level Full TimeColumbus, Ohio12h ago
-
Software Engineer, AI Developer Experience USD 160K-200KAutomation | Cloud automation | Command Line | Command-line Interface | Data Analysis401k | In-home office setup | Internet stipend | Learning & development support | Medical, dental & vision coverageSenior-level Full TimeUnited States - Remote Opportunity R12h ago
-
Senior Manager, AI/ML Engineering USD 180K-200KAPIs | Cloud Platforms | Data Pipelines | Databricks | Enterprise Architecture401k | Adoption leave | Employer Paid Long-term Disability | Employer Paid Short-term Disability | Health insuranceSenior-level Full TimeChicago, Illinois, United States12h ago