ML Engineer

TATA Consulting Services • Cupertino, CA

47d • $100,000 - $140,000

About The Position

Strong experience in machine learning, with a focus on edge AI and lightweight model deployment. Expertise in ML frameworks such as PyTorch, TensorFlow, JAX. Proficiency in programming languages such as C/C++, Python, and experience with ML model optimization. Ability to work independently and collaboratively in a fast-paced startup environment.

Requirements

• Experience in machine learning
• Experience in edge AI
• Experience in lightweight model deployment
• Expertise in ML frameworks such as PyTorch, TensorFlow, JAX
• Proficiency in programming languages such as C/C++, Python
• Experience with ML model optimization
• Ability to work independently
• Ability to work collaboratively in a fast-paced startup environment

Nice To Haves

• Understanding of ML compiler and runtime design
• Experience working with tools such as Optimum, ONNX, TensorRT, TFLite/LiteRT, ncnn, or CoreML
• Familiarity with hardware acceleration techniques
• Experience in embedded system development

What This Job Offers

• Job Type: Full-time
• Career Level: Mid Level
• Industry: Professional, Scientific, and Technical Services
• Education Level: No Education Listed
• Number of Employees: 5,001-10,000 employees