Experience
15 - 22 yrs
Job Location
Bengaluru, India
Vacancy
1
Designation
Machine Learning Engineer
Job Type
ONSITE
Job Description
1) Machine Learning Engineer - Python, PyTorch, JAX or Tensorflow - [ID 013]
Job Description
- Experience in ML frameworks such as PyTorch, JAX or Tensorflow.
Experience with distributed training and distributed training framework such as DeepSpeed.
Experience with LLM or Vision, especially large model is a plus.
Excellent python programing skills, including debugging, profiling, and perf analysis.
Experience with ML pipeline.
Strong communication and problem-solving skills.
Skills:
Python , PyTorch, JAX or Tensorflow
Job Location:
Bengaluru
Experience:
15 - 22 Yrs