Software Engineer · AI/ML Systems, Backend & Distributed Platforms
I build production ML/LLM and agent systems in Python and Julia, and I fix the tools underneath them — with merged contributions to the internals of Flax NNX and the CPython interpreter. Most of my work sits at the boundary between production-grade ML and the messy reality of deploying it in regulated, high-stakes environments: healthcare, enterprise supply chain, and early-stage startups.
| Project | Contribution |
|---|---|
| google/flax | #4 contributor by commits over the trailing 12 months (contributor graph) — 23 commits, +1,165 / −524, all merged into the NNX core. 10 merged PRs fixing module-system, transform (jit/vmap), and dtype-promotion bugs, including #4891 and #5008, both shipped in v0.12.1. |
| python/cpython | 5 merged fixes. Fixed a hard crash in the C text-I/O core — TextIOWrapper.tell() assertion failure on standalone carriage return (#141331), backported to Python 3.13 and 3.14. Plus four merged documentation fixes: re.Match.group() range claim (#144696), asyncio Task cancellation propagation (#141249), BufferedRandom inheritance (#141629), and object comparison semantics (#141221). |
| pytorch/pytorch | Approved core PR (#167209). |
| uber/causalml | Fixed estimation_sample_size not propagating from UpliftRandomForestClassifier to individual trees (#850) — silently broken behaviour in the core uplift Cython layer. |
languages = ["Python", "Julia", "C/C++ (CPython core, Cython)", "SQL", "Bash"] genai_and_agents = [ "LLM agent & tool-use architectures", "LangChain", "LangGraph", "PydanticAI", "MCP servers", "RAG (hybrid dense+sparse retrieval, reranking)", "multi-provider model routing & failover", "structured outputs", "LoRA / SFT fine-tuning", ] ml_and_eval = [ "PyTorch", "JAX/Flax (NNX internals)", "XGBoost", "scikit-learn", "HuggingFace", "DeepEval", "LLM-as-judge", "regression suites", ] backend_and_infra = [ "async/concurrent pipelines", "microservices (REST + message queues)", "blue/green deploys", "distributed tracing (OpenTelemetry)", "GCP (Vertex AI, BigQuery)", "AWS (EC2, Lambda, S3)", "Docker", "GitHub Actions CI/CD", "PostgreSQL", "Redis", "Qdrant", "RabbitMQ", "Airflow", ]
- Website: mohsinmehmood.com
- LinkedIn: mohsinmehmood-m
- Email: [email protected]
- Location: Pakistan · Open to relocation