Software Engineer, Machine Learning, Edge Tensor Processing Unit

8 Hours ago • 2-4 Years • Artificial Intelligence

Job Summary

Job Description

This Software Engineer role focuses on training and optimizing machine learning models for the Tensor Processing Unit (TPU), specifically for edge devices. Responsibilities include developing and training edge-optimized models for various applications (computer vision, natural language processing, speech). The role also involves co-developing model optimization tools and infrastructure, customizing neural network architectures, building efficient ML training pipelines, and creating profiling/visualization tools. Collaboration with researchers and application developers is key. The position requires expertise in machine learning, model optimization, and experience with deep learning frameworks.
Must have:
  • Develop and train edge optimized models
  • Co-develop model optimization tools
  • Work with researchers and developers
  • Build efficient ML training pipelines
  • C++/Python & Data Structures/Algorithms experience
  • 2+ years ML model development experience
Good to have:
  • Master's/PhD in CS with ML focus
  • Experience with TensorFlow, PyTorch, or JAX
  • Publications or research in deep learning
  • 4+ years ML model development experience

Job Details

Minimum qualifications:

  • Bachelor’s degree in Computer Science or related technical fields, or equivalent practical experience.
  • 2 years of experience with software development in C++/python, and with data structures/algorithms.
  • 2 years of experience in developing and training machine learning models.

Preferred qualifications:

  • Master's degree or PhD in Computer Science or related technical field with an emphasis on Machine Learning/Artificial Intelligence.
  • 4 years of experience in developing and training machine learning models.
  • Experience in publications or research related to deep learning.
  • Experience working with any deep learning frameworks, Tensorflow, Pytorch or JAX.

About the job

Google's software engineers develop the next-generation technologies that change how billions of users connect, explore, and interact with information and one another. Our products need to handle information at massive scale, and extend well beyond web search. We're looking for engineers who bring fresh ideas from all areas, including information retrieval, distributed computing, large-scale system design, networking and data storage, security, artificial intelligence, natural language processing, UI design and mobile; the list goes on and is growing every day. As a software engineer, you will work on a specific project critical to Google’s needs with opportunities to switch teams and projects as you and our fast-paced business grow and evolve. We need our engineers to be versatile, display leadership qualities and be enthusiastic to take on new problems across the full-stack as we continue to push technology forward.

As a Software Engineer you will work on training and optimizing machine learning models for the Tensor Processing Unit (TPU). By working on enabling different models spanning varied applications (camera, Speech, Translate, TTS, etc.) on Edge TPU, you will gain exposure and expertise in the exciting direction of efficient model architectures, optimization methods and on-device machine learning activities at Google. Along with your technical expertise, you will manage project priorities, deadlines and deliverables.

Google's mission is to organize the world's information and make it universally accessible and useful. Our team combines the best of Google AI, Software, and Hardware to create radically helpful experiences. We research, design, and develop new technologies and hardware to make computing faster, seamless, and more powerful. We aim to make people's lives better through technology.

Responsibilities

  • Work on developing and training edge optimized models for genAI, computer vision, natural language and speech use cases.
  • Co-develop model optimization tools and infrastructure modules needed for automating customization and training of neural networks and architecture design space exploration.
  • Work with researchers and application developers to customize neural network architectures.
  • Write modular and efficient Machine Learning (ML) training pipelines and assist in building profiling and visualization tools.

Similar Jobs

Canva - Senior Computer Vision Engineer - Photo AI

Canva

London, England, United Kingdom (Remote)
2 Months ago
Playrix - Feature Owner (LiveOps)

Playrix

Montenegro (Remote)
5 Months ago
ByteDance - Software Engineer, Inference

ByteDance

San Jose, California, United States (On-Site)
5 Months ago
Playrix - Game Designer

Playrix

Ireland (Remote)
5 Months ago
Playrix - Game Designer

Playrix

Georgia (Remote)
5 Months ago
N-iX - Senior AI/GenAI Solution Engineer

N-iX

(Remote)
2 Days ago
Rackspace Technology - AI/ML Architect

Rackspace Technology

Vietnam (Remote)
1 Month ago
ByteDance - Student Researcher (Doubao (Seed) - Foundation Model - Generative AI)

ByteDance

Seattle, Washington, United States (On-Site)
3 Weeks ago
Genies - Machine Learning Engineer, Character Animation & Motion AI

Genies

Los Angeles, California, United States (On-Site)
1 Month ago
Google - Product Strategy and Operations Principal, Machine Learning

Google

Sunnyvale, California, United States (On-Site)
9 Hours ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Canva - Senior Computer Vision Engineer - Photo AI

Canva

Vienna, Vienna, Austria (Remote)
1 Month ago
Ello - Tech Lead, Machine Learning

Ello

San Francisco, California, United States (On-Site)
3 Weeks ago
Canva - Senior Machine Learning Engineer - Photo AI

Canva

Vienna, Vienna, Austria (Remote)
2 Months ago
NVIDIA - Senior ASIC Verification Engineer, Coherent High Speed Interconnect

NVIDIA

Santa Clara, California, United States (Hybrid)
1 Week ago
Playrix - Game Director

Playrix

Portugal (Remote)
5 Months ago
G5 Games - 2D UI/UX Artist

G5 Games

(Remote)
2 Days ago
Match Group - Sr. Software Engineer, Generative AI

Match Group

Palo Alto, California, United States (Hybrid)
6 Months ago
Canva - Senior Computer Vision Engineer - Photo AI

Canva

Vienna, Vienna, Austria (Remote)
2 Months ago
PwC - IN-Manager_ Advanced Analytics & ML _D&A_Advisory_Gurgaon

PwC

Gurugram, Haryana, India (On-Site)
5 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Bengaluru, Karnataka, India

Google - Group Product Manager, Google Core

Google

Bengaluru, Karnataka, India (On-Site)
11 Hours ago
Quizizz - ML Engineer

Quizizz

Bengaluru, Karnataka, India (On-Site)
7 Months ago
Rambus - SMTS Verification Engineering

Rambus

Bengaluru, Karnataka, India (Hybrid)
6 Months ago
SciPlay - Project Manager

SciPlay

Bengaluru, Karnataka, India (On-Site)
2 Weeks ago
WebMobril Technologies - 2D - 3D Artist

WebMobril Technologies

Noida, Uttar Pradesh, India (Hybrid)
6 Months ago
Scopely - Game Design Intern

Scopely

Bengaluru, Karnataka, India (Hybrid)
1 Month ago
PwC - Associate - SAP- GDC

PwC

Kolkata, West Bengal, India (On-Site)
6 Months ago
Info Stretch - Senior Engineer

Info Stretch

Chennai, Tamil Nadu, India (On-Site)
5 Months ago
Shaadi - Software Engineer I (Node.js, Python)

Shaadi

Maharashtra, India (On-Site)
6 Months ago

Get notifed when new similar jobs are uploaded

Artificial Intelligence Jobs

Google - Staff Software Engineer, Core Machine Learning, Google Cloud

Google

California, United States (On-Site)
10 Hours ago
ByteDance - Student Researcher (Doubao (Seed) - Foundation Model - Generative AI)

ByteDance

Seattle, Washington, United States (On-Site)
2 Months ago
Google - Research Scientist, Google Cloud AI

Google

Sunnyvale, California, United States (On-Site)
11 Hours ago
Google - Staff Software Engineer, Platforms

Google

Sunnyvale, California, United States (On-Site)
9 Hours ago
Inworld AI - Forward Deployed Engineer (AI Gameplay Engineer)

Inworld AI

Mountain View, California, United States (On-Site)
3 Weeks ago
NVIDIA - Senior Software Engineer - Automated Parallel Programming

NVIDIA

Santa Clara, California, United States (Remote)
2 Months ago
Meta - Software Engineer, Machine Learning

Meta

Menlo Park, California, United States (On-Site)
5 Months ago
Google - Software Engineering Manager, AICore, Applied ML

Google

Taipei City, Taiwan (On-Site)
8 Hours ago
Roofstacks - AI/ML Engineer

Roofstacks

İstanbul, İstanbul, Türkiye (Hybrid)
3 Weeks ago
Spell Brush - LLM Engineer

Spell Brush

San Francisco, California, United States (On-Site)
3 Weeks ago

Get notifed when new similar jobs are uploaded

About The Company

A problem isn't truly solved until it's solved for all. Googlers build products that help create opportunities for everyone, whether down the street or across the globe. Bring your insight, imagination and a healthy disregard for the impossible. Bring everything that makes you unique. Together, we can build for everyone.

Bucharest, Bucharest, Romania (On-Site)

Bengaluru, Karnataka, India (On-Site)

Bengaluru, Karnataka, India (On-Site)

Dublin, County Dublin, Ireland (On-Site)

Bengaluru, Karnataka, India (On-Site)

Sunnyvale, California, United States (On-Site)

Sunnyvale, California, United States (On-Site)

View All Jobs

Get notified when new jobs are added by Google

Level Up Your Career in Game Development!

Transform Your Passion into Profession with Our Comprehensive Courses for Aspiring Game Developers.

Job Common Plug