Ajay Saini

United States

Join Prog.AI to see contacts

Summary

🤩

Rockstar

🎓

Top School

Ajay Saini is an engineering leader and entrepreneur with 11 years of experience building scalable ML systems, from founding a YC-backed startup to managing GenAI training platforms at MosaicML/Databricks. He blends hands-on research in Bayesian and statistical modeling (MIT MEng) with production engineering—contributing to Apache Spark MLlib (including Python-only persistence, UnaryTransformer, and parallel one-vs-rest training) and improving model-training tooling at MosaicML. Ajay has shipped ML-driven features at scale—ransomware detection and distributed data classification at Rubrik—and led teams that operationalize large-model training for customers worldwide. He’s equally comfortable refactoring core training engines and shipping product-facing services, and his background shows a rare mix of academic rigor, patent-backed innovation, and startup grit.

10 years of coding experience

10 years of employment as a software developer

Master of Engineering - MEng Computer Science Concentration: Artificial Intelligence, Master of Engineering - MEng Computer Science Concentration: Artificial Intelligence at Massachusetts Institute of Technology

High School, High School at Acton-Boxborough Regional High School

Github Skills (24)

apache-spark10

pytorch10

spark10

python10

multiple-selection10

feature-selection10

scikit10

testing10

big-data10

machine-learning10

ml10

scala10

deep-learning10

variable-selection10

gridsearchcv10

Programming languages (4)

JavaC++ScalaPython

Github contributions (5)

databricks/spark-sklearn

Jun 2017 - Jun 2017

(Deprecated) Scikit-learn integration package for Apache Spark

Role in this project:

ML Engineer

Contributions:35 commits, 2 PRs, 16 pushes in 7 days

Contributions summary:Ajay primarily contributed to the `spark-sklearn` project, focusing on the implementation and maintenance of the `GridSearchCV` class, a crucial component for hyperparameter tuning within the scikit-learn framework on Apache Spark. Their commits demonstrate efforts to align the library with newer versions of scikit-learn, addressing deprecation warnings, fixing imports, and updating the compatibility of spark. The user also addressed tests and bug fixes, improving functionality, and ensuring the integration of scikit-learn features with Apache Spark.

spark-mlapachesparkscikit-learnscala

mosaicml/composer

Oct 2021 - Apr 2022

Supercharge Your Model Training

Role in this project:

ML Engineer

Contributions:194 reviews, 18 commits, 23 PRs in 5 months

Contributions summary:Ajay's contributions center around refactoring and improving testing within the `composer` repository, a project focused on accelerating machine learning model training. The user made code cleanups and formatting changes. Specific contributions involved test updates and enhancements related to label smoothing algorithms. Additionally, the user made changes to the engine and trainer files, indicating work with the core components of model training.

pytorchml-systemsdeep-learningneural-networksmachine-learning

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