Machine Learning Engineer (TensorFlow + JAX)

5 - 10 years

20 - 25 Lacs

Bengaluru Delhi / NCR Mumbai (All Areas)

Posted:2 months ago| Platform: Naukri logo

Apply

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

Mock Interview

Practice Video Interview with JobPe AI

Start Machine Learning Interview
cta

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

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.

coding practice

Enhance Your Python Skills

Practice Python coding challenges to boost your skills

Start Practicing Python Now

RecommendedJobs for You

Bengaluru, Delhi / NCR, Mumbai (All Areas)

kolkata, mumbai, new delhi, hyderabad, pune, chennai, bengaluru