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, Canonsburg5h ago
-
Senior Manager, Quantitative Consulting USD 149K-248KAI Governance | Agentic Workflows | Artificial Intelligence | Generative AI | Language ModelsFlexible work environment | Total rewards programSenior-level Full TimeUS-NC-Charlotte, US-NY-New York, United States, North …5h ago
-
AI | AI Inference | CPU | Containers | GPUSenior-level Full TimeSeattle, Washington, United States7h ago
-
Machine Learning Engineer - AI Compiler Optimization USD 156K-387KCo-design | Code generation | Compiler optimization | Graph optimization | Hardware-Software Co-designMid-level Full TimeSan Jose, California, United States7h ago
-
Software Engineer, Sensors & Computer Vision USD 146K-208K3D Reconstruction | C++ | Camera | Clustering | Convolutional Neural NetworksMid-level Full TimeSunnyvale, CA8h ago
-
Ablation Studies | Action Recognition | Computer Vision | Data Augmentation | Data PreprocessingSenior-level Full TimeMenlo Park, CA8h ago
-
Principal Engineer, AI Ecosystem USD 307K-428KCUDA | Cloud Native | Cloud Orchestration | Deep learning | Distributed SystemsSenior-level Full TimeSunnyvale, CA, USA; Kirkland, WA, USA8h 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, USA8h ago
-
Bi-Encoder | Cross-Encoder | Data Processing | Data Structures | Data Structures and AlgorithmsBonus | Equity | Health benefitsSenior-level Full TimeSunnyvale, CA, USA; Kirkland, WA, USA8h 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, …8h ago
-
Siri AI Software Engineer, Siri Core USD 161K-282KAPI Design | C++ | Concurrency | GenAI | Language ModelsSenior-level Full TimeCupertino13h 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 States14h ago
-
AI Architect USD 190K-230KAI Foundry | AI Search | Agent workflows | Agents | AzureInclusive benefits | Wellness programsSenior-level Full TimeUnited States17h ago
-
Applied AI Lead-Vice Presient USD 177K-215KAWS | Agent coordination | Agent workflow | Agent workflow orchestration | Agentic designSenior-level Full TimeJersey City, NJ, United States18h 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
-
AI/ML Engineer USD 99K-225KAWS | Azure | Confluent | Deep learning | DockerPaid leave | Professional development | Tuition assistanceMid-level Full TimeUSA, VA, Arlington (1550 Wilson Blvd), …19h ago
-
AI Software Engineer, Senior USD 86K-198KAWS | Access Control | Asynchronous programming | Automated testing | CI/CDDependent care | Disability insurance | Health insurance | Life insurance | Paid leaveSenior-level Full TimeUSA, HI, Joint Base Pearl Harbor …19h 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 HQ20h ago
-
Data Analysis | Experiment design | Geospatial Data | Machine Learning | PyTorch401k | Dental insurance | Health insurance | Office food and beverages | Stock option planMid-level Full TimeRWC HQ20h ago
-
Data provenance | Experimental Design | Geospatial analysis | Machine Learning | Model Evaluation401k | Dental insurance | Health insurance | Office food | Stock optionsMid-level Full TimeRWC HQ20h ago
-
Machine Learning Engineer, Relevance and Personalization USD 166K-195KA/B | A/B Testing | Apache Airflow | Apache Kafka | Apache SparkMid-level Full TimeUnited States20h ago
-
Machine Learning Engineer II USD 160K-231KAlgorithms | Computer Vision | Data Pipelines | Data analytics | Entity recognitionMid-level Full TimeRemote - USA R21h 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 States21h ago
-
Senior-level Full TimeCupertino22h ago