Home
>
Jobs in Bengaluru
>
High Heads
>
Machine Learning Engineer (TensorFlow + JAX)

Machine Learning Engineer (TensorFlow + JAX)

High Heads

5 - 10 years

20 - 25 Lacs

Bengaluru Delhi / NCR Mumbai (All Areas)

Posted:1 day ago| Platform:

Apply

Skills Required

JAX PyTorch Machine Learning Numpy Python

Work Mode

Work from Office

Job Type

Full Time

Job Description

Translating production-grade machine learning models from TensorFlow to JAX. Understanding of TensorFlow, including internals and model structure. Building reproducible ML workflows and ensuring version-controlled development. Required Candidate profile Shd have exp of Machine Learning - 5 yrs OR Python - 5 yrs, PyTorch - 2 yrs, JAX - 1 yr, Numpy - 1 yr. Work location India - Remote job(2 Months contract)- 4 hours per day & minimum 20 hours per week

More Jobs at High Heads

Software Developer (C++, Qt, OpenGL, and 3D Graphics Programming )

Pune

8.0 - 13.0 yrs

INR 15 - 25 Lacs

Autosar Diagnostic Developer

Hyderabad

4.0 - 9.0 yrs

INR 10 - 18 Lacs

Sr. Autosar Diagnostic Developer

Hyderabad

6.0 - 9.0 yrs

INR 12 - 22 Lacs

Machine Learning Engineer (TensorFlow + JAX)

Bengaluru, Delhi / NCR, Mumbai (All Areas)

5.0 - 10.0 yrs

INR 20 - 25 Lacs

Mock Interview

Practice Video Interview with JobPe AI

Start Jax Interview Now

Start Your Job Search Today

Browse through a variety of job opportunities tailored to your skills and preferences. Filter by location, experience, salary, and more to find your perfect fit.

Job Application AI Bot

Apply to 20+ Portals in one click

Download Now

Download the Mobile App

Instantly access job listings, apply easily, and track applications.

High Heads

4 Jobs

Login to

Please Verify Your Phone or Email

Confirm Action

Search

Profile

Upskill and Grow with AI

Machine Learning Engineer (TensorFlow + JAX)