Research Engineer, Pretraining, DeepMind
Tasks
- Focus on inference optimized model design for pretraining
- Optimize JAX code execution on TPUs
- Understand XLA primitives
- Work across pretraining fine tuning and serving pipelines to deliver strong models
Perks/Benefits
- N/A
Skills/Tech-stack
Fine Tuning | Inference Optimization | JAX | Language Models | Large Language Models | Machine Learning | Pretraining | RLHF | Reinforcement Learning | Supervised Learning | TPU | Transformer | XLA
Education
Bachelor of Engineering | Bachelor of Science | Master of Science
Roles
Regions
Countries
States
Cities
Related jobs
-
Senior-level Full TimeUnited States2h ago
-
Senior-level Full TimeUnited States2h ago
-
Computational Biologist - Protein Engineering USD 150K-250KAWS | Amazon Web Services | CUDA | Conda | Deep learningRelocation supportEntry-level Full TimeSan Francisco, CA, US3h ago
-
Senior Google Cloud Data Engineer USD 170K-240KAlerting | Anomaly Detection | Apache Beam | Apache PubSub | Batch Data ProcessingCareer advancement | Dental insurance | Health savings account | Hybrid work options | Medical insuranceSenior-level Full TimeNew York City5h ago
-
Machine Learning Engineer Intern (E-Commerce Knowledge Graph - CV/Multimodal/NLP) - 2026 Summer (BS/MS) USD 122K-243KEntity Linking | Graph Modeling | Knowledge graphs | Language Processing | Machine LearningDevelopment workshops | Social eventsEntry-level InternshipSeattle, Washington, United States5h ago
-
Senior Software Engineer, AI Coding Tools USD 244K-588KDeep learning | Fine Tuning | GPU | Inference acceleration | Language ModelsSenior-level Full TimeSan Jose, California, United States6h ago
-
Checkpointing | Data-Driven Optimization | Data-driven | Distributed Training | Fault ToleranceMid-level Full TimeSan Jose, California, United States6h ago
-
Machine Learning Engineer Intern (E-commerce-Supply Chain & Logistics) - 2026 Summer (MS) USD 136K-237KClassification | Data Mining | Deep learning | Forecasting | Knowledge graphsEntry-level InternshipSeattle, Washington, United States6h ago
-
Software Engineer, Generative AI, Workspace USD 147K-211KC++ | Distributed Systems | Generative AI | Information Retrieval | Integration TestingBenefits | Bonus | EquityMid-level Full TimeBoulder, CO, USA6h ago
-
Staff Software Engineer, Machine Learning, Google Chat USD 207K-300KAgentic Workflows | Caching | Cloud Spanner | Continuous Delivery | Continuous integrationSenior-level Full TimeSunnyvale, CA, USA7h ago
-
Software Engineer III, AI/ML Computer Vision, AR USD 147K-211KC++ | Computer Vision | Data Processing | Data Structures | Data Structures and AlgorithmsSenior-level Full TimeSan José, CA, USA7h ago
-
Agentic AI | C plus plus | C# | Cloud services | Data ProcessingMid-level Full TimeSan Francisco, CA, USA7h ago
-
Forward Deployed Engineer I, GenAI, Google Cloud USD 102K-145KAPIs | Agent systems | Cloud Computing | Cloud platform | Fine TuningSenior-level Full TimeSan Francisco, CA, USA; Calgary, AB, …7h ago
-
Applied AI Engineer, Agentic Systems USD 115K-192K.NET | APIs | Anthropic | CrewAI | Evaluation FrameworksAI and productivity tools access | Remote work accessSenior-level Full TimeRemote - United States R11h ago
-
Supply Chain Data Engineer Ii USD 94K-118KDBT | Data Governance | Data Modeling | Data Pipelines | Data Quality401k | Disability insurance | Employee stock purchase plan | Health insurance | Life insuranceMid-level Full TimeWayne, PA, US, 1908712h ago
-
AI/ML Engineer 2 USD 101K-165KAI Agents | API Development | AWS | Azure | CI/CDDisability insurance | Family leave | Flexible spending accounts | Life and AD D Insurance | Medical/Dental/Vision insuranceSenior-level Full TimePhiladelphia, PA, US, 1910312h ago
-
Staff AI/ML Engineer USD 240K-270KAWS | Agentic Workflows | Cloud platform | Data Curation | Deep learning401k | Commuter benefits | Dog-friendly office | Equity | FSA benefitsSenior-level Full TimeSan Francisco, CA12h ago
-
Senior Analytics Engineer, RegTech USD 150K-180KAWS | Acceptance Testing | Agile | Airflow | DBT401k match | Company-provided phone | Extended leave | Health insurance | Observed holidaysSenior-level Full TimeNew York, New York, United States15h ago
-
Applied AI ML Engineer-Vice President USD 150K-210KAWS Bedrock | AWS SageMaker | Amazon EKS | AutoPrompt | DDPBackup childcare | Financial coaching | Health care coverage | Mental health support | On-site health and wellness centersExecutive-level Full TimeNew York, NY, United States17h ago
-
Applied AI ML Engineer-Senior Associate USD 175K-210KAWS Bedrock | AWS SageMaker | Amazon EKS | Containerization | Data PreprocessingSenior-level Full TimeNew York, NY, United States17h ago
-
AI Engineer USD 157K-283KAPI Integration | Agentic Workflows | Autogen | Cloud infrastructure | CrewAISenior-level Full TimeUSA MD Columbia (Field), United States18h ago
-
API Integration | Agent Orchestration | Amazon Bedrock | Angular | Autogen401k plan | Commuter benefits | Disability benefits | Life insurance | Paid time offExecutive-level Full Time110832-NY-30 Hudson Yards, New York, United …18h ago
-
Senior Machine Learning Engineer - AI/ML USD 148K-247KAWS | Airflow | Apache Flink | Apache Spark | Argo WorkflowsContinual development | Flexible work environment | Health and wellness benefits | Internal career growth | Paid relocation orientation processSenior-level Full TimeSan Mateo HQ, United States18h ago
-
Analytics Lead, Data Analytics & Modeling USD 107K-140KData Architecture | Data Governance | Data Modeling | Data Pipelines | Data TransformationHybrid workSenior-level Full TimeMalvern, PA, United States18h ago
-
Manager, Data Science USD 60K-75KAI | AWS | Cloud services | Machine Learning | PyTorchHybrid work modelMid-level Full TimeMalvern, PA, United States18h ago