Jas Bali is an experienced ML and data engineer with nine years building production-grade machine learning systems, primarily on the Databricks platform. A lead SSA at Databricks and developer for the Databricks Labs automl-toolkit, he combines hands-on SDE skills with deep ML lifecycle expertise. Jas is an active MLflow contributor who improved autologging for TensorFlow/Keras—adding input example and signature logging and cross-version test compatibility—helping make a widely used open-source project more robust for TensorFlow workflows. Based in Atlanta, he brings practical experience across ML engineering, data engineering, and software development, consistently shipping tooling that bridges research and production. Colleagues would note his focus on observable, reproducible models and a knack for smoothing integration pain points that are often invisible in deployments.
The open source AI engineering platform for agents, LLMs, and ML models. MLflow enables teams of all sizes to debug, evaluate, monitor, and optimize production-quality AI applications while controlling costs and managing access to models and data.
Role in this project:
ML Engineer
Contributions:50 reviews, 6 commits, 13 PRs in 10 months
Contributions summary:Jas's commits primarily focus on enhancing the autologging capabilities of the MLflow library, specifically for TensorFlow and Keras models. They added input example and signature logging for tf.keras models and tf.estimator, enabling better tracking of model inputs and outputs. The contributions involved modifying unit tests to ensure cross-version compatibility, adding support for dataset input types, and refining the logging behavior for various model types. This work improves the usability and functionality of MLflow for TensorFlow-based machine learning workflows.
Open source platform for the machine learning lifecycle
Contributions:2 PRs, 105 pushes, 13 branches in 11 months
pythonlifecycledeep-learningmlmachine-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.