Machine Learning Engineer (Computer Vision)
Lytx Inc.Jul, 2023 - Present2 yr 9 months
Leveraged cloud platforms like GCP and AWS, conducted regular model training and benchmarking experiments. Utilized multiple GPUs with multiprocessing techniques to accelerate model training and optimize compute resource usage. Enhanced training efficiency through data and model parallelism in frameworks like TensorFlow and PyTorch. Developed and implemented an Mnasnet model with SSD head from scratch, achieving lightweight performance for passenger counting in vehicles. Led model training, fine-tuning, and benchmarking of advanced computer vision models, including DinoV2 and Detectron2 for object detection tasks. Preprocessed and managed large-scale datasets for deep learning pipelines using SQL and cloud buckets. Ran model benchmarking experiments on curated data sets, optimizing models through hyperparameter tuning and evaluating metrics such as accuracy, precision, recall, F1 score, ROC-AUC, and latency. Conducted a proof of concept for Neural Architecture Search on GCP Vertex AI with Google, optimizing model architecture and latency using search spaces like Spinenet and Mnasnet.