About Google
Learn more about Google. Explore our innovative AI products and services, and how we. Backed by Kleiner Perkins.
About the role
At Google DeepMind our mission is to build the world's first general-purpose learning agent. Central to this mission is the complex task of measuring the intelligence of our prototypes. As a Software Engineer, you will be working with the cutting edge AI agents developed by our exceptional team of Machine Learning and Neuroscience research scientists. Your responsibilities will include everything from creating systems for agent testing using 2D and 3D games to developing test problems within physics simulators. Found on 1752vc Careers, the job board for startup and VC roles.
What they're looking for
- Bachelor's degree in Computer Science, Software Engineering, Mathematics, or a related STEM field or equivalent practical experience
- 5 years of experience in software development, systems programming, or applied ML engineering or equivalent education credit (e.g., internships, student research, pre-graduate working experience)
- 3 years of experiencewith performance optimization, systems data analysis, visualization tools, or debugging
- 3 years of experience with Generative AI, Large Language Models (LLMs), Machine Learning and related frameworks
- 1 year of experience designing and launching production software products or foundation AI systems
More about this role
- Drive down the cost per token of Gemini inference by optimizing throughput, accelerator utilization, batching, scheduling, and KV cache efficiency. Achieve these gains through expert optimization and by building agentic workflows that enable AI agents to discover, implement, and validate performance improvements at scale.
- Be among the first to bring Gemini to new TPU generations by characterizing hardware at launch, tuning kernels, sharding, and serving configurations for specific memory hierarchies and interconnects, and delivering production-ready performance. Share these insights to influence future TPU hardware and compiler roadmaps.
- Implement model compression techniques—such as quantization and sparsity—to reduce the cost-per-token for production Gemini models. Evaluate new architectures, optimizers, and training methods to achieve target quality with reduced compute requirements.
- Bachelor's degree in Computer Science, Software Engineering, Mathematics, or a related STEM field or equivalent practical experience.
- 5 years of experience in software development, systems programming, or applied ML engineering or equivalent education credit (e.g., internships, student...
Browse similar: AI jobs · AI startup jobs · Startup jobs