AI Research Engineer, Pre-Training
Tasks
- Build and optimize model training parallelism
- Co design models with researchers
- Develop CUDA kernels
- Develop training data loading pipelines
- Enhance model networking performance
- Implement fault tolerance for training
- Improve kernel performance
- Improve large scale model training
- Shape research agenda
Perks/Benefits
- N/A
Skills/Tech-stack
ASIC | CUDA | CUDA Graphs | Deep learning | FPGA | JAX | Language Models | Large Language Models | PyTorch | Triton | XLA
Education
N/A
Regions
Countries
States
Cities
Related jobs
-
AI Intern USD 50K-58KAccess Control | Artificial Intelligence | Cybersecurity | Data Privacy | Help deskEntry-level InternshipYork, United States9h ago
-
Machine Learning Engineer USD 131K-178KAWS | Cassandra | Convolutional Neural Networks | Data Lakes | Data PipelinesMid-level Full TimeRemote, NY, US R9h ago
-
APIs | Android Development | Artificial Intelligence | IOS Development | JavaMid-level ContractChicago, United States9h ago
-
Amazon S3 | Data Engineering | Data Modeling | Data Pipelines | Data QualitySenior-level Full TimeNew York11h ago
-
Amazon S3 | Automation | Data Engineering | Data Modeling | Data Pipelines401k match | Dental insurance | Life insurance | Long-term disability | Medical insuranceSenior-level Full TimePrinceton11h ago
-
Senior Databricks Forward Deployed Engineer - GPS USD 119K-198KAPI Integration | AWS | Airflow | Azure | CI/CDTravelSenior-level Full TimeArlington/Rosslyn, Virginia, United States; Atlanta, Georgia, …11h ago
-
Lead AI and Data Solutions Engineer II USD 137K-229KAmazon Web Services | Apache Spark | Application Programming | Application Programming Interfaces | Cloud ComputingSenior-level Full TimeSacramento, California, United States; Tempe, Arizona, …11h ago
-
Software Engineer, Systems ML - SW/HW Co-design USD 117K-173KAI infrastructure | Bias Mitigation | C# | C++ | Co-designSenior-level Full TimeSunnyvale, CA | Redmond, WA12h ago
-
Software Engineer, Machine Learning USD 213K-293KAPI Design | Agent Orchestration | Artificial Intelligence | Bias Mitigation | C++Senior-level Full TimeSunnyvale, CA | Remote, US | … R12h ago
-
Senior Software Engineer, Generative AI, Google Ads USD 174K-252KComputer Vision | Data Processing | Debugging | GenAI | Information RetrievalSenior-level Full TimeMountain View, CA, USA12h ago
-
C++ | Data Processing | Debugging | Deep learning | Few-Shot LearningSenior-level Full TimeMountain View, CA, USA12h ago
-
Senior Software Engineer, Generative AI USD 174K-252KAgent-based | Agent-based systems | Cloud platform | Data Structures | Data Structures and AlgorithmsSenior-level Full TimeSunnyvale, CA, USA12h ago
-
Software Engineer III, Generative AI, Payments Risk USD 147K-211KAgent systems | Algorithms | Analytics | Big Data | Computer VisionSenior-level Full TimeMountain View, CA, USA12h ago
-
C++ | Data Analysis | Data Processing | Deep learning | EmbeddingsSenior-level Full TimeMountain View, CA, USA12h ago
-
Machine Learning Research Engineer USD 146K-222KData Analysis | Data Visualization | Deep learning | GPU Programming | Graph Neural Networks401k | Education reimbursement program | Flexible benefits package | Flexible schedule | Relocation assistanceMid-level Full TimeLivermore, CA, United States19h ago
-
Senior Machine Learning Engineer USD 229K-360KAB Testing | AWS SageMaker | Airflow | Amazon S3 | Apache FlinkDisability benefits | Equity awards | Health insurance | Life insurance | Paid time offSenior-level Full TimeSan Jose, California19h ago
-
Member of Technical Staff, Robotics Research Engineer USD 270K-370KData collection | Deep learning | Demonstration data | Diffusion Models | JAXSenior-level Full TimeNew York22h ago
-
Staff Engineer, Software Embedded Applications (R5033) USD 128K-189KArchitecture Reviews | Avionics | BSP | C# | C++Senior-level Full TimeUnited States23h ago
-
Gen AI Engineering Analyst - Vice President USD 113K-170KAWS | Accuracy | Apache Kafka | Apache Spark | Azure401k | Accident insurance | Disability insurance | Life insurance | Medical, dental, and vision coverageExecutive-level Full Time14000 CITI CARDS WAY BUILDING C …23h ago
-
Principal AI/ML Engineer USD 165K-226KC# | C++ | CI/CD | CUDA | Computer Vision401k match | Dental insurance | Health insurance | Life insurance | Paid time offSenior-level Full TimeRemote PA - PA PAR, United … R23h ago
-
Senior AI Engineer USD 74K-147KAI Builder | API Development | AWS | Azure | Azure MLFlexible remote work policy | Flexible work-life balance | Knowledge sharing | Professional development | Supportive environmentSenior-level Full TimeChicago, United States23h ago
-
Senior Agentic AI Engineer USD 83K-203KArtificial Intelligence | Azure OpenAI | Cloud Computing | Code review | Data PipelinesDental insurance | Medical insurance | Paid time off | Retirement savings options | Vision insuranceSenior-level Full TimeWork At Home-Texas, United States23h ago
-
Machine Learning Research Engineer USD 99K-225KBenchmarking | Code review | Computer Vision | Conformal Prediction | Contrastive LearningPaid leave | Professional development | Tuition assistanceMid-level Full TimeUSA, VA, Springfield (7500 Geoint Dr), …23h ago
-
Agile | C++ | Deep learning | Distributed Computing | GPU ComputingDiscretionary bonus | Flexible time off | Healthcare | Leave benefits | Retirement benefitsExecutive-level Full TimeNY7 - 50 Hudson Yards, New … R23h ago
-
Associate Director - AI Engineering USD 150K-190KAI Governance | AWS | Agentic Frameworks | Agentic framework | Amazon SageMakerCompany-matched student loan contribution | Continuous learning | Family-friendly benefits | Financial wellness programs | Flexible time offMid-level Full TimeUS - NY NYC - 55 …23h ago