Lytx - Machine Learning Engineer - PyTorch/Tensorflow (4-6 yrs)

As our Machine Learning Engineer, you will join our Applied Machine Learning Team in India, which develops machine learning and computer vision algorithms to monitor and assess the state of drivers and their environments to identify risks and improve safety for our clients. You will contribute to all aspects of the DL/ML model development and deployment cycle to improve the accuracy and throughput at reduced latency to help enhance and differentiate us as the leader in the Video Safety and Telematics industry. If this sounds like you, we encourage you to apply!

- End-to-end development of DL/ML models - prototyping, formulation, data engineering, model training, etc.

- Implement, debug, and improvise DL/ML models.

- Develop and optimize cloud/Edge deployment of DL models.

- Optimize DL models - quantization, model pruning, etc.

- Other duties as assigned.

Requirements :

- Bachelor's/MS degree in CS, EE, EC, or related fields.

- 4+ years of experience in prototyping a given problem, surveying the literature, and coming up with suitable methods and training the models using one of the DL/ML frameworks.

- Strong understanding of core ML concepts, and hands-on experience with at least 2 DL frameworks like PyTorch/TF.

- Strong analytical and problem-solving skills with fundamental knowledge of relevant math, probability, and algorithms.

- Experience in computer vision problems such as detection, classification, and segmentation. Additional experience with non-vision problems is a plus.

- Ability to work under minimal supervision and run structured experiments with the end goal in mind.

- Ability to work in collaboration with developers on model implementation to ensure adequate system performance (latency and accuracy), security, scalability, and availability.

- Familiarity with AWS Services (EC2 Sagemaker, DynamoDB, ECR, Lambda, etc. ) is preferred.

- Experience with MLOps tools such as SageMaker, Airflow, MLFlow, H2O, etc., is a plus.

- Experience with SQL or similar DB query language.

- Experience working with large datasets / long-running / distributed training.

- Experience with any low-level programming language like C/C++ or a deeper understanding of system architecture.

- A strong teammate who enjoys working in a collaborative, fast-paced team-focused environment.