Software Engineer, Machine Learning, Edge Tensor Processing Unit

1 Week 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

NVIDIA - Senior ASIC Verification Engineer, Coherent High Speed Interconnect

NVIDIA

Hsinchu, Hsinchu City, Taiwan (On-Site)
1 Month ago
NVIDIA - Senior Deep Learning Research Engineer, Advanced AI Systems

NVIDIA

Santa Clara, California, United States (On-Site)
1 Month ago
Trend Micro - NLP / Prompt Engineer (VicOne_Automotive Security)

Trend Micro

Taipei City, Taiwan (On-Site)
6 Months ago
G5 Games - 2D UI/UX Artist (Hidden objects project)

G5 Games

Yerevan, Yerevan, Armenia (Remote)
3 Months ago
Playrix - Game Director

Playrix

Serbia (Remote)
6 Months ago
ByteDance - Senior Machine Learning Engineer

ByteDance

San Jose, California, United States (On-Site)
1 Week ago
Google - Staff Software Engineer, AI/ML GenAI, Google Ads

Google

Mountain View, California, United States (On-Site)
1 Week ago
ByteDance - Research Scientist in Foundation Model, Music Core Machine Learning Graduates - 2024 Start (PhD)

ByteDance

San Jose, California, United States (On-Site)
5 Months ago
NVIDIA - Deep Learning Software Engineer, Performance Optimization

NVIDIA

Tokyo, Japan (On-Site)
3 Months ago
Zoox - Senior/Staff Software Engineer - Simulator

Zoox

Foster City, California, United States (Hybrid)
6 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Playrix - Feature Owner (LiveOps)

Playrix

Ireland (Remote)
6 Months ago
NVIDIA - Senior Software Engineer, RTL Optimization Tools

NVIDIA

Santa Clara, California, United States (On-Site)
2 Weeks ago
Playrix - Game Director

Playrix

Georgia (Remote)
6 Months ago
Google - Senior Customer Solutions Engineer, gTech Sustainability

Google

Sunnyvale, California, United States (On-Site)
4 Days ago
NVIDIA - Senior Deep Learning Research Engineer, Advanced AI Systems

NVIDIA

Santa Clara, California, United States (On-Site)
1 Month ago
G5 Games - 2D UI/UX Artist

G5 Games

Astana, Astana, Kazakhstan (Remote)
1 Week ago
PwC - IN-Manager_ Advanced Analytics & ML _D&A_Advisory_Gurgaon

PwC

Gurugram, Haryana, India (On-Site)
6 Months ago
Madison Logic - Sr. Data Scientist

Madison Logic

Pune, Maharashtra, India (On-Site)
6 Months ago
Playrix - Game Designer

Playrix

(Remote)
6 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Bengaluru, Karnataka, India

Luxoft - Senior DevOps Engineer (Azure)

Luxoft

New Delhi, Delhi, India (Remote)
5 Months ago
Google - Software Engineer III, AI/ML

Google

Hyderabad, Telangana, India (On-Site)
6 Days ago
Nagarro - Principal Consultant, Agile

Nagarro

India (Remote)
6 Months ago
ION - Analyst - LCM - Mumbai - 764

ION

Mumbai, Maharashtra, India (On-Site)
6 Months ago
Info Stretch - Senior Engineer

Info Stretch

Bengaluru, Karnataka, India (On-Site)
5 Months ago
Simple Viral Games - Android Developer Intern

Simple Viral Games

Bengaluru, Karnataka, India (On-Site)
8 Months ago
Hitachi - FO Technical Consultant

Hitachi

Pune, Maharashtra, India (On-Site)
6 Months ago
Velotio Technologies - Senior Software Engineer (Golang)

Velotio Technologies

Maharashtra, India (Remote)
1 Month ago
Warner Bros Games - Principal Engineer- Backend (MSC Team), Bangalore

Warner Bros Games

Bengaluru, Karnataka, India (On-Site)
9 Months ago
Nagarro - Senior Staff Engineer, Delivery ETIL

Nagarro

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

Get notifed when new similar jobs are uploaded

Artificial Intelligence Jobs

Zoox - Software Engineering  - Returnship

Zoox

Foster City, California, United States (Hybrid)
6 Months ago
Hitachi - Artificial Intelligence - JBU

Hitachi

Chennai, Tamil Nadu, India (On-Site)
6 Months ago
Microsoft - Senior Applied Researcher

Microsoft

Redmond, Washington, United States (On-Site)
5 Days ago
Henkel - Data Scientist-Intern

Henkel

Pune, Maharashtra, India (On-Site)
7 Months ago
ByteDance - Software Engineer (Applied Machine Learning - Enterprise)

ByteDance

San Jose, California, United States (On-Site)
1 Month ago
Lionbridge Games - Language AI Specialist (Test & Tech)

Lionbridge Games

Masovian Voivodeship, Poland (On-Site)
2 Months ago
Meta - Research Scientist Intern, Smart Glasses in Wearables AI (PhD)

Meta

Menlo Park, California, United States (On-Site)
5 Months ago
Scopely - Senior AI Creative (Motion) - Monopoly Go

Scopely

Barcelona, Catalonia, Spain (Hybrid)
1 Month ago
Google - Software Engineer, PhD, Early Career, Campus, Machine Learning, Systems and Cloud AI, 2025 start

Google

Sunnyvale, California, United States (On-Site)
4 Months ago
Corsair - Senior Manager, AI & Data

Corsair

Munich, Bavaria, Germany (On-Site)
1 Month 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.

Fremont, California, United States (On-Site)

Mountain View, California, United States (On-Site)

Bengaluru, Karnataka, India (On-Site)

Dublin, County Dublin, Ireland (On-Site)

Atlanta, Georgia, United States (On-Site)

Ho Chi Minh City, Ho Chi Minh City, Vietnam (On-Site)

Seattle, Washington, 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