AI Training Infrastructure Engineer
Tasks
- Build automation and tools for AI developers
- Build distributed training infrastructure
- Collaborate on distributed computing and platform readiness
- Design systems for training reliability
- Diagnose training throughput stability and cost issues
- Establish training infrastructure best practices
- Implement fault tolerance checkpointing recovery
- Improve training efficiency and resource utilization
- Integrate training systems into production pipelines
- Scale training workloads across GPU clusters
Perks/Benefits
Skills/Tech-stack
Checkpointing | DeepSpeed | Distributed Computing | Fault Tolerance | Fine Tuning | GPU clusters | Human Feedback | Kubernetes | Machine Learning | MegatronLM | Performance optimization | Production Machine Learning | PyTorch | Python | Ray | Recovery | Reinforcement Learning | Resource utilization | Supervised Fine Tuning
Education
N/A
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 R12d ago
-
Featured Feat. Associate Director, Data Labs USD 167K-167KAWS | Cloud Computing | Compute Infrastructure | Data Analysis | LLM GovernanceConference speaking opportunities | Hybrid work schedule | Media appearancesSenior-level Full TimeWashington, District of Columbia, 20004, United … R25d ago
-
Associate Cloud Data Engineer II USD 78K-97KAgile | Airflow | Apache Beam | BigQuery | Cloud ComposerOnsite workMid-level Full TimeAda, MI, United States, 493552h ago
-
Staff Software Engineer USD 114K-178KAI Safety | AWS | AWS Bedrock | Agentic Systems | AirflowHealth insurance | Paid time off | Retirement contributionsSenior-level Full TimeSt. Louis, Missouri, US, 63103-23553h ago
-
Senior Cell Modeling & Analytics Engineer USD 140K-175K3D Modeling | Circuit modeling | D Q D V | Data Analysis | Data Visualization401k plan | Equity opportunities | Flexible time off | Medical, dental, and vision coverageSenior-level Full TimeBroomfield, Colorado or Burlingame, California7h ago
-
Airflow Design | Analysis Techniques | Availability | Compute Integration | Cooling systemsOnsite work | Travel required | Valid Driver License | Work long hours and weekendsSenior-level Full TimeBastrop, TX9h ago
-
Artificial Intelligence | Automation | LLM integrations | Langchain | Language ModelsContractor | Fully remoteSenior-level ContractLos Angeles R9h ago
-
Forward Deployed AI Engineer (GenAI, AWS) USD 175K-200KAWS | AWS CDK | AWS Neptune | Amazon ECS | Amazon SageMakerRemote-friendly cultureSenior-level Full TimeNew Jersey9h ago
-
Staff Data Engineer USD 200K-260KApache Airflow | Apache Beam | Apache Spark | BigQuery | DBT401k plan | Dental insurance | Disability insurance | Equity | Flexible time offSenior-level Full TimeSunnyvale, CA9h ago
-
[C] Data Engineer, Safeguards USD 380K-485KAccess Control | Amazon Kinesis | Amazon Redshift | Apache Airflow | Apache SparkFlexible work schedule | Flexible working hours | Generous vacation | Parental leaveSenior-level Full TimeSan Francisco, CA | New York …10h ago
-
Embedded Test Automation Engineer USD 130K-150KAPIs | CI/CD | Cloud Databases | Disk partitioning | Embedded Linux401k match | Health, dental, and vision insurance | Paid Holidays | Paid parental leave | Paid sick timeSenior-level Full TimeHayward, CA10h ago
-
Lead Machine Learning Engineer USD 172K-276KAmazon Web Services | Azure | CI/CD | Cloud platform | DatabricksLearning and development programs | Mentorship and coachingSenior-level Full TimeChicago, Illinois, USA R11h ago
-
Robotics Engineer USD 170K-221KC++ | Camera Object Detection | Computer Vision | Gazebo | Isaac Sim401k matching | Background check clearance | Dental insurance | Drug Test Clearance | Life insuranceEntry-level Contract Full TimeSunnyvale, California11h ago
-
Imagery Product Engineer II- AI/Deep Learning USD 79K-133KAmazon AWS | C# | CI/CD | Databricks | Deep learning401k | Dental insurance | Health insurance | Life insurance | Paid HolidaysMid-level Full TimeRedlands, CA11h ago
-
Solution Engineer USD 165K-216KAnalytics | Cloud Computing | Data Lake | Data Warehousing | Data fabricSenior-level Full TimeUS-FL-Remote R12h ago
-
Software Engineer, ML Data USD 180K-230KAWS | Apache Hive | Apache Spark | ClickHouse | GolangDental insurance | Equity | Health insurance | Hybrid work | Team gatherings once per quarterSenior-level Full TimeSan Francisco Bay Area12h ago
-
Senior-level Full TimeSan Francisco, California, United States12h ago
-
Senior-level Full TimeNew York, New York, United States12h ago
-
Solution Engineer USD 138K-181KAI machine learning | Analytics | Cloud Computing | Data Warehouse | ETLSenior-level Full TimeUS-CO-Denver12h ago
-
Solution Engineer USD 138K-181KData Warehousing | Generative AI | Language Models | Language Processing | Large Language ModelsMid-level Full TimeFins Only-US-CA-San Mateo12h ago
-
Accelerator programming | Caching | Command Line | Command-line Tools | Data PipelinesCompetitive benefits | Flexible working hours | Generous vacation | Optional equity donation matching | Parental leaveSenior-level Full TimeSan Francisco, CA | New York …12h ago
-
AI Software Engineer USD 127K-175KAPI Development | Agent Orchestration | Agent runtimes | Agent systems | C++Dental insurance | Employee assistance program | Flexible Paid Vacation | Flexible paid sick leave | Flexible spending accountEntry-level Full TimeSan Francisco, CA12h ago
-
Staff Robotics System and Control Engineer USD 164K-282KC plus plus | Collision Avoidance | Collision Checking | Computer Vision | Constrained motion401(k) plan match | Annual bonus | Commuter benefits | Dental insurance | Disability insuranceSenior-level Full TimeMountain View, USA12h ago
-
Senior Data Engineer USD 126K-180KAmazon Kinesis | Apache Airflow | Apache Beam | Apache Flink | Apache Spark401k match | Bonus | Equity grant | Flexible time off | Health plansSenior-level Full TimeNew York, New York; Miami, Florida; … R12h ago
-
Lead Algorithm & Signal Processing Engineer USD 157K-220KAdaptive Signal Processing | Bayesian Methods | C plus plus | Deep learning | Electronic WarfareTop Secret clearanceSenior-level Full TimeDayton, OH13h ago