Software Engineer on MTIA Compiler and Performance Team.

Working on PyTorch FX Compiler for MTIA Silicon.

Promoted to Machine Learning Engineer II

• Currently working on the XLA compiler for the Neuron backend

• Middle end compiler loop optimizations

• High performance kernels targeting Neuron hardware

Machine Learning Engineer on AWS Neuron Training Team - Annapurna Laboratories

• High performance deep learning on Trn1 and Inf2 instances

• OpenXLA compiler passes

• PJRT support for Neuron backend

• Training on multi thousand Neuron device clusters.

• GSPMD with PyTorch/Jax for automatic parallelization of machine learning computations.

• Distributed training for Bedrock

• Pretraining and fine tuning multi hundred billion parameter large language models on Neuron cores

• Training using XLA and PyTorch and Jax

https://aws.amazon.com/machine-learning/neuron/

Software Engineer on AWS Bedrock Training Team

• Pretraining and fine tuning large language models (Titan models)

• Distributed multi node training using Deepspeed and NeMo

https://aws.amazon.com/bedrock/

Researching with Dr. Panagiotis Apostolellis on the development of using chatbots for depression screening. Currently using Google DialogFlow, Kommunicate and Swift to develop Marcus: Depression Screening.