Open to opportunities

Shruti Jain

@shrutijain2

Message

Experienced Deep Learning Software Engineer specializing in kernel optimization.

India

Message

What I'm looking for

I am seeking a role that fosters innovation and collaboration, allowing me to leverage my deep learning expertise while contributing to impactful projects.

I am a dedicated Deep Learning Software Engineer with extensive experience in optimizing machine learning kernels for large-scale AI models. Currently, I am a Member of Technical Staff-2 at Cerebras Systems, where I have successfully optimized kernels for Mixtral models, achieving a remarkable 2.5x speedup and enhancing performance for complex models like GPT-J and Llama.

Previously, I worked as a Senior Deep Learning Engineer at Intel, where I developed a Python-based execution framework that supported multiple backends, including PyTorch and TensorFlow. My contributions to the Nervana I-1000 chip enabled it to achieve full ONNX compliance, and I was recognized with the Amazing Work Award for my collaborative efforts. My technical expertise spans across various hardware and software tools, allowing me to tackle complex challenges in the field of deep learning.

Experience

Work history, roles, and key accomplishments

Current

Member of Technical Staff-2

Current

Cerebras Systems

Apr 2023 - Present (2 years 1 month)

As a Member of Technical Staff-2 at Cerebras Systems, I optimized kernels for large-scale AI models, achieving significant performance improvements. My work included debugging complex runtime issues and enhancing kernel stability across various models. I also developed test infrastructure to streamline kernel validation processes.

Kernel Optimization Debugging Performance Tuning TensorFlow PyTorch Optimization Unit Testing

Sr. Deep Learning Engineer

Intel Inc.

May 2018 - May 2022 (4 years)

In my role as a Senior Deep Learning Engineer at Intel, I developed a Python-based execution framework for various ML frameworks and ensured ONNX compliance for Nervana chips. I also created Linux drivers and performance measurement tools, contributing to the advancement of deep learning technologies.

Python Onnx PyTorch TensorFlow Kernel Enablement Collaboration

Graduate Technical Intern

Intel Inc.

Jun 2017 - May 2018 (11 months)

During my internship at Intel, I developed an OS-agnostic audio driver framework and built a topology XML parser tool. My contributions led to recognition for innovation and a pre-placement offer.

Audio Driver Development XML Innovation OS Agnostic Framework Collaboration

Education

Degrees, certifications, and relevant coursework

Manipal Institute of Technology

Master of Technology, Computer Science and Engineering

2016 - 2018

Grade: 8.35/10

Completed a Master of Technology in Computer Science and Engineering with a focus on advanced computing techniques and machine learning applications.

Shri Vaishnav Institute of Technology

Bachelor of Technology, Computer Science and Engineering

2011 - 2015

Grade: 8.02/10

Earned a Bachelor of Technology in Computer Science and Engineering, gaining foundational knowledge in software development and engineering principles.