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
-
Generative AI Engineer III - State and Local Government USD 110K-218KArtificial Intelligence | Data Analysis | Data Pipelines | Generative AI | Language ModelsMentorship | Professional development | Travel opportunitiesSenior-level Full TimeAustin, Texas, United States; Boston, Massachusetts, …13h ago
-
Attention Mechanism | DPO | Data Generation | Data Preprocessing | Deep learningEntry-level InternshipCalifornia, Sunnyvale14h ago
-
Bandit Algorithms | Data Analysis | Deep learning | Generative AI | Language ModelsCareer growth opportunities | Hands-on project experience | Research mentorshipEntry-level InternshipSan Jose, California, United States14h ago
-
Machine Learning Engineer Graduate (TikTok Trust and Safety-CV/NLP/Multimodal LLM) - 2026 Start (PhD) USD 136K-259KActive Learning | Computer Vision | Distributed Training | Language Models | Language ProcessingEntry-level Full TimeSeattle, Washington, United States14h ago
-
Staff Software Engineer, Generative AI USD 207K-301KC++ | Computer Vision | Data Processing | Data Structures | Data Structures and AlgorithmsSenior-level Full TimeSunnyvale, CA, USA15h ago
-
Staff Software Engineer, Google Cloud Agentic AI USD 207K-301KAgentic AI | Artificial Intelligence | C++ | Compliance | Distributed ComputingBonus | Equity | Health insurance | Paid time off | Retirement planSenior-level Full TimeNew York, NY, USA15h ago
-
AI Research Engineer USD 100K-150KAblation Studies | Accelerator hardware | Agentic Systems | Data Quality | Data labeling100 percent remote work | Career growth opportunities | Visa transfer support for qualified candidatesMid-level Full TimeUnited States - Remote R1d ago
-
Quantitative Analytics Professional USD 105K-157KAgentic Workflows | Apache Spark | Generative AI | Language Models | Large Language ModelsMid-level Full TimeHeadquarters 4, United States1d ago
-
LLM Engineer USD 100K-150KAdapter-Tuning | Automated benchmarking | DPO | Dataset curation | Direct Preference OptimizationCareer growth potential | Remote workMid-level Full TimeUnited States - Remote R1d ago
-
Sr Lead Machine Learning Engineer USD 229K-262KBias Variance | CI/CD | Dask | Data Pipelines | Deep learningSenior-level Full TimeMcLean, VA, United States1d ago
-
Principal Data Science Engineer - Financial Crimes USD 107K-216KAnomaly Detection | CI/CD | Clustering | DBT | Decision TreesSenior-level Full Time499 Washington Blvd., Jersey City NJ, …1d ago
-
Director, GenAI Technology USD 126K-255KClustering | Data Engineering | Data Visualization | Deep learning | Inference OptimizationEducational assistance | Emotional well-being support | Health care coverage | Learning resources | Paid time offExecutive-level Full Time245 Summer St, Boston MA, United …1d ago
-
Audio Machine Learning | Audio Separation | Cloud Computing | Fine Tuning | Language ModelsAviation immersion opportunities | Equity upside | Free flight training | Remote work | Technical leadership opportunitySenior-level Full TimeSan Francisco, CA; Onsite R1d ago
-
Senior MLOps & Generative AI Engineer - Remote USD 91K-152KAWS | Alerting | Azure | CI/CD | Deep learningAdoption reimbursement | Emergency backup care | Fertility and surrogacy reimbursement | Long-term disability | Medical/Dental/VisionSenior-level Full TimeCorp Facilities MPB - 350 Centre … R1d ago
-
Senior AI Engineer - Agentic Workflow USD 151K-332KAWS | Agent Orchestration | Anthropic Claude | Docker | GPT4Community support | Health benefits | Hybrid work environment | In person option | Remote work optionSenior-level Full TimeSeattle (WA), United States1d ago
-
Lead Software Engineer, Gen AI USD 128K-208KAPI Gateway | AWS | AWS Lambda | Amazon RDS | Amazon SageMaker401k savings plans | Adoption benefits | Career development | Disability benefits | Employee assistance programSenior-level Full TimeChicago, Illinois, United States1d ago
-
AI Engineer - Advisor Software USD 89K-158KAPIs | Amazon Web Services | C# | C++ | CI/CD401k | Caregiver leave | Dental insurance | Flexible work schedule | Health insuranceMid-level Full TimeUnited States of America Milwaukee (South …1d ago
-
Staff Applied Scientist USD 244K-320KAgentic Systems | Artificial Intelligence | Benchmarking | CI/CD | Computer VisionEmployee communities | Experience bonus | Hybrid work model | Wellness reimbursementSenior-level Full TimeSeattle, Washington, United States1d ago
-
AI Research Engineer USD 190K-280KDeep learning | Generative AI | Language Models | Language Processing | Large Language ModelsCareer development | Diversity and inclusion | Flexible work environmentMid-level Full TimeSeattle, Washington, United States; South San …1d ago
-
Sr. AI Engineer (Applied AI & ML Systems) USD 132K-165KAgentic AI | Context engineering | Continuous Improvement | Data Engineering | Data PipelinesE learning license | Hackathons | Healthcare benefits | Home office setup allowance | Identity theft protectionSenior-level Full TimeUnited States R1d ago
-
Machine Learning Engineer USD 159K-216KC++ | Computer Vision | ITK | Image analytics | Image registrationEntry-level Full TimeSunnyvale, CA, United States1d ago
-
AWS | Azure | Causal Inference | Cloud platform | DatabricksBackup childcare | Financial coaching | Health care coverage | Mental health support | On-site wellness centersExecutive-level Full TimeNew York, NY, United States1d ago
-
FWI & AI Scientist USD 110K-140KAccelerated Forward Modeling | Cloud Computing | Forward Modeling | Full waveform inversion | GPU ComputingMid-level Full TimeHouston, United States1d ago
-
A/B | A/B Testing | AWS | Airflow | Amazon Redshift401k matching | Employee assistance program | Flexible time off | Flexible work arrangement | Paid HolidaysMid-level Full TimeRemote, US R1d ago
-
Practice Manager - AI & Data USD 160K-190KAWS | Agent systems | Amazon SageMaker | Apache Spark | AzureCareer growth opportunities | Comprehensive benefits | MentorshipSenior-level Full TimeBroomfield, CO. Greensboro, NC. Troy, Michigan1d ago