profile-pic

Pranit Abhinav

I am a motivated and inquisitive machine learning engineer with 5 years of experience in building and optimizing distributed machine learning systems for model training and inference.
  • Role

    Machine Learning Engineer (Computer Vision)

  • Years of Experience

    4 years

Skillsets

  • Keras
  • Vertex AI
  • Transformers
  • time series forecasting
  • SQL
  • SocketIO
  • Sklearn
  • Sagemaker
  • Rnn
  • Reinforcement Learning
  • rag
  • ONNX
  • MATLAB
  • LSTM
  • LLM
  • Python
  • JavaScript
  • Java
  • Graph Algorithms
  • GCP
  • Data Analysis
  • Cnn
  • Azure
  • AWS
  • ANN
  • Computer Vision
  • Flask
  • TensorFlow
  • PyTorch
  • NLP

Professional Summary

4Years
  • Jul, 2023 - Present2 yr 9 months

    Machine Learning Engineer (Computer Vision)

    Lytx Inc.
  • Jun, 2022 - Jun, 20231 yr

    Machine Learning Engineer II

    Raven Protocol Pte. Ltd.
  • Jun, 2021 - Jun, 20221 yr

    Machine Learning Engineer I

    Raven Protocol Pte. Ltd.
  • Jun, 2019 - Jul, 20201 yr 1 month

    Machine Learning Engineer

    Microsoft AI for Earth

Applications & Tools Known

  • icon-tool

    Keras

  • icon-tool

    Vertex

  • icon-tool

    SQL

  • icon-tool

    MATLAB

  • icon-tool

    Socket.IO

  • icon-tool

    Amazon SageMaker

  • icon-tool

    Rnn

  • icon-tool

    ONNX

  • icon-tool

    LSTM

  • icon-tool

    Flask

  • icon-tool

    TensorFlow Hub

Work History

4Years

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.

Machine Learning Engineer II

Raven Protocol Pte. Ltd.
Jun, 2022 - Jun, 20231 yr
    Developed and built a full peer-to-peer decentralized and distributed deep learning training system as a component of the Ravenverse framework to simultaneously update weights and communicate gradients among compute providers utilizing P2P servers. Developed and implemented techniques and architecture for splitting large functional AI models and their weights into smaller submodels and computational graphs to support distributed training on low compute resources. Demonstrated training large AI models using applications like neural style transfer model on Ravenverse for style transfer image generation. Worked on key functionalities necessary for the Ravenverse distribution framework. Demonstrated large AI models distributed training using imagenette dataset for architectures like resnet-50 and VGG.

Machine Learning Engineer I

Raven Protocol Pte. Ltd.
Jun, 2021 - Jun, 20221 yr
    Recognized and implemented many math operations required for Deep Learning computations in Requester and Provider libraries of Ravop and Ravpy. Worked on benchmarking contributors' devices using payloads for scheduling operations to a contributor based on its computing power. Contributed to Ravens centralized distribution server Ravsock for distributing ML tasks and operations. Documented several of Ravens distributed computing libraries including Ravop, Ravml, Ravdl, and RavJS. Implemented various machine learning algorithms, loss functions, and metric functions for Ravens ML libraries supporting distributed training.

Machine Learning Engineer

Microsoft AI for Earth
Jun, 2019 - Jul, 20201 yr 1 month
    Worked under Prof. Mohd Anul Haq as the sole undergraduate student on deep learning algorithms for GIS applications like hyperspectral analysis and groundwater forecasting using Azure compute credits. Trained and tuned ML models and presented findings as part of the AI for Earth conference.

Major Projects

5Projects

Hyperspectral Image Classification on NASAs AVIRIS NG dataset

    Used various Deep Learning models like CNN and LSTM to classify Hyperspectral images from ISRO and NASAs AVIRIS NG dataset utilizing Azure compute credits. Trained and compared multiple architectures like FDSSC and LSTM models, generating benchmark results on preprocessed Hdf5-format datasets.

Ground Water Level Monitoring using NDVI data

    Trained and compared LSTM, XGBOOST, SVM, and DNN models to monitor and predict groundwater levels in Rajasthan using NDVI and rainfall data. Results were presented at the Microsoft headquarters in Redmond, USA.

Pong with pixels using reinforcement learning

    Trained an AI agent using Deep Q-networks, policy gradient methods, and adversarial training to play Pong using pixel data from the OpenAI Gym environment.

Neural machine translation on a parallel text dataset

    Implemented a Neural Machine Translation model and trained on a parallel English-to-Hindi text dataset using a transformer architecture.

Chinese checkers using Reinforcement learning and minimax trees

    Designed an adversarial zero-sum game between two AI agents using minimax trees and function approximation with weight tuning and alpha-beta pruning for decision-making.

Education

  • Btech. in Computer Science

    NIIT University (2021)