Gautam Mittal

San Francisco, California, United States

Join Prog.AI to see contacts

Summary

🤩

Rockstar

🎓

Top School

Gautam Mittal is a Member of Technical Staff in San Francisco with 13 years of experience building ML-driven systems, HPC training pipelines, and backend/devops tooling. He has driven research and production work across startups and big tech—from founding research engineer at Contextual AI scaling retrieval-augmented systems to ML research at Google Brain and production perception models at Tesla advised by Andrej Karpathy. Gautam is a hands-on engineer who bridges research and engineering: he contributes to JAX/Flax and Magenta on generative and differentiation tooling, and to SkyPilot’s backend and CLI to simplify running AI workloads across clouds and Kubernetes. He’s skilled in end-to-end model lifecycle and infra (pretraining, retrieval, evaluation, deployment, and cost-aware GPU orchestration) and has helped grow teams and hiring programs at early-stage ventures. Notably, his open-source work includes practical CLI features (interactive sessions, SSH/SCP/Rsync) that improve developer experience for large-scale ML runs.

13 years of coding experience

6 years of employment as a software developer

BS Electrical Engineering and Computer Science, BS Electrical Engineering and Computer Science at University of California, Berkeley

Henry M. Gunn High School

MS Computer Science, MS Computer Science at Stanford University

English, Japanese

Github Skills (27)

cloud-manager10

python10

cloud-computing10

command-line-interface10

machine-learning10

automatic-differentiation10

manage10

tensorflow10

encoding10

data-processing10

yaml10

command-line10

encode10

jax10

devops10

Programming languages (12)

TypeScriptCSSJinjaShellC++CJavaScriptObjective-C

Github contributions (5)

skypilot-org/skypilot

Oct 2021 - Aug 2022

SkyPilot: Run AI and batch jobs on any infra (Kubernetes or 15+ clouds). Get unified execution, cost savings, and high GPU availability via a simple interface.

Role in this project:

Backend & DevOps Engineer

Contributions:2 reviews, 79 commits, 1 PR in 10 months

Contributions summary:Gautam primarily contributed to the backend infrastructure and CLI tooling, specifically focusing on the "sky run" command for launching tasks from YAML configurations. Their work involved fixing execution commands and supporting YAML task configurations by modifying core files related to task management, cloud backend integration, and configuration templating. The user also added to the CLI by introducing cluster handles, interactive sessions, and SSH/SCP/Rsync support, improving the user experience.

pythonspot-instancesserverlesstpuml-infrastructure

google/flax

Dec 2020 - Feb 2022

Flax is a neural network library for JAX that is designed for flexibility.

Role in this project:

ML Engineer

Contributions:1 review, 14 commits, 5 PRs in 1 year 2 months

Contributions summary:Gautam primarily contributed to the `flax` repository by modifying and extending the `early_stopping` module. Their changes included refactoring the module to use a dataclass, fixing style issues, adding test cases and improving its usability. The work involved adapting the module to fit the project's needs and improving the overall quality of the existing code by adding new tests.

deep-learningneural-networksneural-networkflaxjax

Find and Hire Top DevelopersWe’ve analyzed the programming source code of over 60 million software developers on GitHub and scored them by 50,000 skills. Sign-up on Prog,AI to search for software developers.

Request Free Trial