About
👋

About

Who am I

My name is Kejie Zhang.
I am a software engineer.
I am currently working as a deep learning compiler engineer in AWS Neuron team.

Work Experience

notion image
 

ML Compiler Engineer, AWS Neuron Team

Jan 2025 - Present Cupertino, CA
 
notion image

ML Compiler Engineer, Sambanova Systems

Sep 2021 - Jan 2025 Palo Alto, CA
  • 🛠️ Developed Deep Learning Compiler
notion image

Software Engineer, Audible/Amazon

Feb 2020 - Sep 2021, Cambridge, MA
notion image

Software Engineer Intern, 6 River Systems

July 2018 - Dec. 2018, Waltham, MA
  • 🤖 Worked in Fulfillment Service Team focus on Warehouse Robotics Job Allocation

More about me

I'm fascinated by the challenges in Machine Learning, High-Performance Computing, and Distributed Systems. I thrive on tackling difficult problems, learning new things, and building something that makes an impact.
 
Cheers!😄