Skip to content
View aj-prime's full-sized avatar
3 followers · 0 following
  • Microsoft

Achievements

Achievement: Pair ExtraordinaireAchievement: Pull Sharkx2

Achievements

Achievement: Pair ExtraordinaireAchievement: Pull Sharkx2

Block or report aj-prime

Block user

Prevent this user from interacting with your repositories and sending you notifications. Learn more about blocking users.

You must be logged in to block users.

Maximum 250 characters. Please don't include any personal information such as legal names or email addresses. Markdown supported. This note will be visible to only you.
Report abuse

Contact GitHub support about this user’s behavior. Learn more about reporting abuse.

Report abuse
aj-prime/README.md

Hi there 👋

Hi, I’m Arpan Jain, a Senior Researcher at Microsoft working on deep learning inference for Large Language Models within the AI Frameworks team. I earned my Ph.D. in High-Performance Deep Learning from The Ohio State University, where I developed novel parallelization strategies for distributed DNN training. My work spans across Deep Learning traning and inference frameworks like LBANN, MVAPICH2, and DeepSpeed, with publications in top HPC and AI conferences. I enjoy building scalable AI systems and have previously worked on speech analysis, time series modeling, and computer vision. I’m also passionate about mentoring and sharing knowledge through teaching and community engagement in HPC-AI forums.

Popular repositories Loading

  1. pytorch pytorch Public

    Forked from pytorch/pytorch

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

    C++

  2. dlrm dlrm Public

    Forked from facebookresearch/dlrm

    An implementation of a deep learning recommendation model (DLRM)

    Python

  3. FlexFlow FlexFlow Public

    Forked from flexflow/flexflow-train

    A distributed deep learning framework that supports flexible parallelization strategies.

    Cuda

  4. Elemental Elemental Public

    Forked from llnl/Elemental

    Distributed-memory, arbitrary-precision, dense and sparse-direct linear algebra, conic optimization, and lattice reduction

    C++

  5. lbann lbann Public

    Forked from LBANN/lbann

    Livermore Big Artificial Neural Network Toolkit

    C++

  6. examples examples Public

    Forked from pytorch/examples

    A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

    Python