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
-
AI Governance | AI guardrails | API Integration | Agent systems | AutoGPTLocal onsite days | Remote work 3 daysMid-level Full TimeAustin, TX R3h ago
-
AI Evaluation | AI Governance | AI guardrails | Agent systems | Agent workflowsSenior-level Full TimeAustin, TX R3h ago
-
Mid-level Full TimeAustin, TX R3h ago
-
Airflow | Apache Spark | CI/CD | Data Governance | Data LakehouseOnsite work required | Relocation possibleSenior-level Full TimeAustin, TX3h ago
-
ArcGIS | Data Engineering | Data Governance | Data Pipelines | Data QualitySenior-level Full TimeBoston, NY, United States3h ago
-
AWS Bedrock | CI/CD | CUI | Data Governance | Data PipelinesGovernment furnished equipment | Pacific Time core hours support | Remote workEntry-level Full TimeMenlo Park, CA, United States3h ago
-
Mid-level Full TimeTel Aviv-Yafo, Tel Aviv District, IL R4h ago
-
Senior Research Engineer / Scientist -AI for Databases USD 207K-368KAWS | Architecture Search | Artificial Intelligence | Automated Model Tuning | C++401k match | Dental insurance | Disability coverage | Health insurance | Life insuranceSenior-level Full TimeSeattle, Washington, United States R4h ago
-
Data Engineer (Multiple Positions) USD 148K-300KBig Data | ClickHouse | Data Modeling | Data Query | Data Query Optimization401k savings plan | Dental insurance | Life insurance | Long-term disability | Medical insuranceEntry-level Full TimeSeattle, Washington, United States5h ago
-
Machine Learning System Scheduling Engineer Graduate (Applied Machine Learning) - 2027 Start USD 128K-256KAlgorithms | CUDA | Cloud Storage | Computer Science | ContainerizationEntry-level Full TimeSan Jose, California, United States5h ago
-
AI/LLM Network Software Development Engineer Graduate (High Speed Network) - 2027 Start (PhD) USD 162K-316KAI network | AI network optimization | C# | C++ | Computer Networking401k with company match | Dental insurance | Life insurance | Long-term disability | Medical insuranceEntry-level Full TimeSan Jose, California, United States5h ago
-
ARM Cortex | ARM Cortex M4 | AV1 | AVC | C++401-k match | Dental insurance | Life insurance | Long-term disability | Medical insuranceEntry-level Full TimeSan Diego, California, United States5h ago
-
Data Lake Infrastructure & Data Analytics Research Engineer Intern (AML-Ark-US) - 2027 Summer USD 86K-86KAlgorithms | Big Data | Cloud infrastructure | Columnar Storage | Data GovernanceHealth insurance | Housing allowance | Life insurance | Paid Holidays | Paid sick timeEntry-level InternshipSeattle, Washington, United States5h ago
-
Research Engineer, Data Lake Infrastructure & Data Analytics Graduate (AML-Ark-US) - 2027 Start (PhD) USD 153K-300KBatch Processing | Columnar Storage | Data Governance | Data Lake | Data ProcessingEntry-level Full TimeSeattle, Washington, United States5h ago
-
AUC Optimization | Apache Flink | Apache Hadoop | Apache Spark | Big Data401k savings plan | Dental insurance | Life insurance | Long-term disability | Medical insuranceEntry-level Full TimeSan Jose, California, United States5h ago
-
Algorithms | CUDA | Computer Architecture | Data Structures | Deep Learning Training401k match | Life insurance | Long-term disability | Medical/Dental/Vision insurance | Paid HolidaysEntry-level Full TimeSan Jose, California, United States5h ago
-
Apache Doris | Apache Flink | Apache Hive | Apache Kafka | Apache Spark401k matching | Dental insurance | Life insurance | Long-Term Disability coverage | Medical insuranceEntry-level Full TimeSan Jose, California, United States5h ago
-
Machine Learning Engineer USD 170K-250KAzure ML | Azure ML Endpoints | Azure Machine Learning | Bayesian Methods | Boosted TreesSenior-level Full TimeAustin, Texas | Fremont, California | …5h ago
-
Data Structures | Data Structures and Algorithms | Deep learning | Ensemble learning | Graph theory401k match | Dental insurance | Health insurance | Life insurance | Long-term disabilityEntry-level Full TimeSeattle, Washington, United States5h ago
-
Deep Neural Networks | Ensemble learning | Graph theory | Hadoop | HiveHealth insurance | Housing allowance | Life insurance | Paid Holidays | Paid sick timeEntry-level InternshipSeattle, Washington, United States5h ago
-
Data Structures | Data Structures and Algorithms | Deep learning | Ensemble Methods | Graph theory401k match | Dental insurance | Life insurance | Long-term disability | Medical insuranceEntry-level Full TimeSeattle, Washington, United States5h ago
-
Data Querying | Data Structures | Deep Neural Networks | Ensemble learning | Graph theory401k matching | Dental insurance | Life insurance | Long-term disability | Medical insuranceEntry-level Full TimeSan Jose, California, United States5h ago
-
Data Structures | Data Structures and Algorithms | Deep Neural Networks | Ensemble trees | Graph theoryHealth insurance | Housing allowance | Life insurance | Paid Holidays | Paid sick timeEntry-level InternshipSan Jose, California, United States5h ago
-
AWS | Azure | Azure ML | C++ | CUDAEntry-level Full TimeSeattle, Washington, United States5h ago
-
Research Intern (AI/LLM Network) - 2026 Start (PhD) USD 114K-114KAI network | AI network optimization | C++ | Computer Networking | Congestion ControlHealth insurance | Housing allowance | Life insurance | Paid Holidays | Paid sick timeEntry-level InternshipSeattle, Washington, United States5h ago