Nikhil Gupta is a Senior Software Engineer in the ML Infrastructure Software team, specializing in High-Performance Computing (HPC) and optimizing machine learning workloads for ArmĀ® Neoverseā¢ servers. He works on integrating advanced performance enhancements into leading ML frameworks like PyTorch, TensorFlow and JAX focusing on developing production-level modules for neural network optimization. His work enables efficient, scalable, and high-performance GenAI deployments on ArmĀ® infrastructure.
