Research Engineer, Pretraining, DeepMind
Tasks
- Focus on inference optimized model design
- Understand XLA primitives and JAX on TPUs
- Work across pretraining fine tuning and serving stack
Perks/Benefits
- N/A
Skills/Tech-stack
Artificial Intelligence | Fine Tuning | Inference Optimization | JAX | Language Models | Large Language Models | Machine Learning | Model Design | Model Serving | Pretraining | TPU | XLA
Education
Bachelor of Arts | Bachelor of Engineering | Bachelor of Science
Roles
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 … R19d ago
-
Data Engineer USD 110K-165KCause analysis | Data Analysis | Data Pipelines | Data Processing | Data TransformationSenior-level Full TimeFAIRFAX, VA, United States4h ago
-
Azure Data -AI -Architect USD 119K-258KAzure Compliance | Azure Data | Azure Data Factory | Azure Databricks | Azure DevOpsSenior-level Full TimeUSA6h ago
-
Mid-level Full TimeNew York, NY, US7h ago
-
AWS | AWS EMR | AWS Glue | Amazon S3 | Apache AirflowSenior-level Full TimePrinceton9h ago
-
Alerting | Data Governance | Data Modeling | Data Quality | Data Versioning401k match | Dental insurance | Life insurance | Long-term disability | Medical insuranceSenior-level Full TimePrinceton9h ago
-
Software Engineer, Machine Learning, Travel Ads USD 147K-211KA/B | A/B Testing | Adbrain | B testing | C++Mid-level Full TimeMountain View, CA, USA10h ago
-
Algorithms | Backend software | C++ | Control Plane | Data StructuresSenior-level Full TimeSunnyvale, CA, USA10h ago
-
Material Science Research Engineer, DeepMind USD 147K-211KCalculus | Data Analysis | Deep learning | JAX | Linear AlgebraMid-level Full TimeMountain View, CA, USA10h ago
-
Research Engineer, Humanoids, DeepMind USD 147K-211KC++ | Debugging | Foundation Models | JAX | Language ProcessingMid-level Full TimeMountain View, CA, USA10h ago
-
Staff Software Engineer, Beyond Live, DeepMind USD 207K-301K2D Games | 3D Games | Audio Tokenization | Audio video pacing | Audio/VideoSenior-level Full TimeNew York, NY, USA; Mountain View, …10h ago
-
Senior Staff Engineer, GDC AI Inference Platform USD 262K-365KAgent coordination | C++ | Cloud infrastructure | Containerization | Data StructuresBonus | Equity | Health insurance | Paid time off | Retirement plansSenior-level Full TimeSunnyvale, CA, USA; Kirkland, WA, USA10h ago
-
Sr AI Engineer USD 115K-142KAgent systems | Agentic Workflows | CI/CD | Citation Grounding | Cloud ComputingSenior-level Full TimeAda, MI, United States, 4935514h ago
-
PIIM Analytics Engineer USD 86K-132KAPIs | Access Control | Azure Databricks | Data Architecture | Data GovernanceDental insurance | Medical insurance | Optional Disability Insurance | Paid time off | Supplemental insuranceMid-level Full TimeClay, NY, US16h ago
-
Senior-level Full TimeSeattle, Washington, United States; USA - …17h ago
-
Principal AI Engineering Architect USD 224K-308KAWS | Data Processing | Docker | Google Cloud | High AvailabilityAdoption and surrogacy reimbursement | Cancer Care Program | Employee Assistance Program (EAP) | Fertility coverage | Flexible time offSenior-level Full TimeUnited States - Remote R17h ago
-
Staff Robotics Engineer, Maritime USD 254K-336KC++ | Collision Avoidance | Computer Vision | Computer Vision Algorithms | Machine LearningHealth benefits | Mentorship | Recovery BenefitsSenior-level Full TimeQuincy, Massachusetts, United States18h ago
-
Senior Robotics Engineer, Maritime USD 220K-292KC++ | Cameras | Computer Vision | Gazebo | LidarHealth insurance | Mentorship | Travel opportunitiesSenior-level Full TimeQuincy, Massachusetts, United States18h ago
-
Machine Learning Engineer, GPU Kernel and Runtime USD 213K-263KCUDA | Deep learning | GPU Computing | Kernel optimization | Machine LearningSenior-level Full TimeMountain View, California20h ago
-
Distributed Systems | Fault Tolerance | GPU Computing | JAX | Low LatencyBonus program | Company benefits program | Equity incentive planSenior-level Full TimeMountain View, California20h ago
-
RFML Engineer, EW USD 166K-220K5G | C++ | Channel equalization | Channel estimation | Deep learningEmployee benefits package | Travel | U.S. Secret security clearance supportMid-level Full TimeCosta Mesa, California, United States21h ago
-
Lead Software Engineer, Data Cloud USD 172K-260KAWS | Azure | Big Data | Big data processing | Containers401k | Employee stock purchasing program | Life and disability insurance | Medical, dental, and vision insurance | Mental health supportSenior-level Full TimeWashington - Bellevue, United States21h ago
-
AI workflows | APIs | Artificial Intelligence | Data Infrastructure | Distributed SystemsDirect collaboration with executive leadership | High impact ownership | Rapid Career Growth Opportunities | Remote-first environment | Visa sponsorshipMid-level Full TimeSan Francisco, California, CA, United States R21h ago
-
AI Scientist USD 98K-123KData Pipelines | Inference services | Language Models | Large Language Models | Machine LearningCareer development | Competitive compensation and benefits | Global opportunities | Pay transparencyMid-level Full TimeAtlanta, GA, United States, United States21h ago
-
Principal, AI Solutions USD 123K-161KAI Automation | Advanced Analytics | Artificial Intelligence | Cloud Computing | Data platformsSenior-level Full TimeReston Town Center, United States21h ago