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
-
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 … R18d ago
-
Agentic AI Engineer USD 101K-151KAgentic Workflows | Debugging | Embeddings | Git | GitHubFinancial benefits | Health benefits | Wellness benefitsMid-level Full TimePennsylvania, Canonsburg4h ago
-
AI | AI Inference | CPU | Containers | GPUSenior-level Full TimeSeattle, Washington, United States6h ago
-
Software Engineer, Sensors & Computer Vision USD 146K-208K3D Reconstruction | C++ | Camera | Clustering | Convolutional Neural NetworksMid-level Full TimeSunnyvale, CA7h ago
-
Ablation Studies | Action Recognition | Computer Vision | Data Augmentation | Data PreprocessingSenior-level Full TimeMenlo Park, CA7h ago
-
Principal Engineer, AI Ecosystem USD 307K-428KCUDA | Cloud Native | Cloud Orchestration | Deep learning | Distributed SystemsSenior-level Full TimeSunnyvale, CA, USA; Kirkland, WA, USA7h ago
-
AI Engineer, Advanced Solutions Lab, Google Cloud USD 152K-222KApache Beam | Apache Hive | Artificial Intelligence | C++ | Cloud ComputingSenior-level Full TimeChicago, IL, USA; Atlanta, GA, USA7h ago
-
Bi-Encoder | Cross-Encoder | Data Processing | Data Structures | Data Structures and AlgorithmsBonus | Equity | Health benefitsSenior-level Full TimeSunnyvale, CA, USA; Kirkland, WA, USA7h ago
-
Business Data Scientist, Applied Machine Learning, GCS USD 138K-198KA/B | A/B Testing | B testing | Causal Inference | Causal MLMid-level Full TimeMountain View, CA, USA; New York, …7h ago
-
Siri AI Software Engineer, Siri Core USD 161K-282KAPI Design | C++ | Concurrency | GenAI | Language ModelsSenior-level Full TimeCupertino12h ago
-
Applied AI ML Lead - Generative AI and Semantic Modeling USD 177K-215KAgent-based | Agent-based AI | Context engineering | Continuous integration | Data ContractsBackup childcare | Financial coaching | Health care coverage | Mental health support | Retirement savings planSenior-level Full TimeJersey City, NJ, United States13h ago
-
AI Architect USD 190K-230KAI Foundry | AI Search | Agent workflows | Agents | AzureInclusive benefits | Wellness programsSenior-level Full TimeUnited States16h ago
-
Applied AI Lead-Vice Presient USD 177K-215KAWS | Agent coordination | Agent workflow | Agent workflow orchestration | Agentic designSenior-level Full TimeJersey City, NJ, United States17h ago
-
Batch Processing | Bias Mitigation | C++ | Data Processing | Experimentation401k employer match | Caregiving support | Family planning support | Flexible vacation | Gender-affirming careSenior-level Full TimeRemote - United States R18h ago
-
Machine Learning Research Engineer USD 140KComputer Vision | Data Assimilation | Data Engineering | Data Pipelines | Data cleaning401k | Dental insurance | Health insurance | Office food and beverages | Stock option planMid-level Full TimeRWC HQ19h ago
-
Data Analysis | Experiment design | Geospatial Data | Machine Learning | PyTorch401k | Dental insurance | Health insurance | Office food and beverages | Stock option planMid-level Full TimeRWC HQ19h ago
-
Data provenance | Experimental Design | Geospatial analysis | Machine Learning | Model Evaluation401k | Dental insurance | Health insurance | Office food | Stock optionsMid-level Full TimeRWC HQ19h ago
-
Machine Learning Engineer, Relevance and Personalization USD 166K-195KA/B | A/B Testing | Apache Airflow | Apache Kafka | Apache SparkMid-level Full TimeUnited States19h ago
-
Machine Learning Engineer II USD 160K-231KAlgorithms | Computer Vision | Data Pipelines | Data analytics | Entity recognitionMid-level Full TimeRemote - USA R20h ago
-
Senior AI Software Engineer USD 92K-209KAPI Development | Agent coordination | Agentic AI | Distributed Systems | Experimentation401k match | Adoption Assistance | Commuter & Parking Benefits | Employee stock purchase plan | Flexible spending accountsSenior-level Full TimeNashville, TN, United States20h ago
-
Senior-level Full TimeCupertino21h ago
-
Forward Deployed AI Engineer USD 90K-110KAPI Integration | AWS | Access Management | Agentic AI | Automated testingEntry-level Full TimeChicago, IL, United States21h ago
-
AIML - Machine Learning Researcher, MLR USD 175K-312KDeep learning | Experiment design | Linear Algebra | Machine Learning | PyTorchSenior-level Full TimeCupertino; San Francisco21h ago
-
Intern, AI Engineer (3 Months, Full-Time) USD 50K-50KAPI Integration | Agentic AI | Chunking | Code Automation | FastAPI401k with company match | Career advancement | Employee assistance program | Flexible time off | Health savings accountEntry-level Full Time InternshipScottsdale, Onsite21h ago
-
Avionics Embedded Software Engineer II USD 106K-150KAutonomous Systems | Bare Metal | Board Support | Board Support Package | BootloaderIn-office work requirementMid-level Full TimePlaya Vista, California, United States22h ago