Skip to main contentA logo with &quat;the muse&quat; in dark blue text.

Software Engineer, Machine Learning, Edge Tensor Processing Unit

AT Google
Google

Software Engineer, Machine Learning, Edge Tensor Processing Unit

Bangalore, India

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.

Want more jobs like this?

Get jobs in Bangalore, India delivered to your inbox every week.

By signing up, you agree to our Terms of Service & Privacy Policy.

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.

Client-provided location(s): Bengaluru, Karnataka, India
Job ID: Google-76351911959110342
Employment Type: Other

Perks and Benefits

  • Health and Wellness

    • Health Insurance
    • Dental Insurance
    • Vision Insurance
    • Life Insurance
    • Short-Term Disability
    • Long-Term Disability
    • FSA
    • HSA
    • Fitness Subsidies
    • On-Site Gym
    • Mental Health Benefits
    • Health Reimbursement Account
    • HSA With Employer Contribution
  • Parental Benefits

    • Birth Parent or Maternity Leave
    • Non-Birth Parent or Paternity Leave
    • Fertility Benefits
    • Adoption Assistance Program
    • Family Support Resources
    • Adoption Leave
  • Work Flexibility

    • Hybrid Work Opportunities
  • Office Life and Perks

    • Commuter Benefits Program
    • Casual Dress
    • Pet-friendly Office
    • Snacks
    • Some Meals Provided
    • On-Site Cafeteria
  • Vacation and Time Off

    • Paid Vacation
    • Paid Holidays
    • Personal/Sick Days
    • Leave of Absence
    • Volunteer Time Off
  • Financial and Retirement

    • 401(K) With Company Matching
    • Company Equity
    • Performance Bonus
    • Financial Counseling
  • Professional Development

    • Tuition Reimbursement
    • Internship Program
    • Learning and Development Stipend
  • Diversity and Inclusion

    • Employee Resource Groups (ERG)

Company Videos

Hear directly from employees about what it is like to work at Google.