Usama Baloch

Machine Learning Engineer at Unify

Karachi Division, Sindh, Pakistan
email-iconphone-icongithub-logolinkedin-logotwitter-logostackoverflow-logofacebook-logo
Join Prog.AI to see contacts
email-iconphone-icongithub-logolinkedin-logotwitter-logostackoverflow-logofacebook-logo
Join Prog.AI to see contacts

Summary

🤩
Rockstar
🎓
Top School
Usama Baloch is a Machine Learning Engineer with five years’ experience building and optimizing ML systems, currently contributing to Unify’s Ivy project to unify ML frameworks. He develops backend features and implements core operations and TensorRT support for Ivy, benchmarks HuggingFace models across targets, and applies sparsification techniques like quantization and pruning to improve performance. Usama also fine-tunes LLMs with LoRA/PEFT and vLLM for domain tasks and has worked on GC and autotuner components, showing a blend of low-level engine work and applied model optimization. Based in Karachi, he pairs academic grounding from the National University of Computer and Emerging Sciences with hands-on open-source contributions—adding PyTorch front-end functions and backend logic for JAX and TensorFlow—to help make framework interoperability practical. Notably, his contributions touch both framework internals and real-world deployment tooling, a combination that accelerates moving SOTA models from research to production.
code5 years of coding experience
bookBachelor's degree, Computer Science, Bachelor's degree, Computer Science at National University of Computer and Emerging Sciences
languagesEnglish, Urdu
stackoverflow-logo

Stackoverflow

Stats
11reputation
232reached
0answers
1question
github-logo-circle

Github Skills (12)

unit-testing10
pytorch10
python10
machine-learning9
deep-learning9
neural-network8
converter8
jax7
tensorflow7
apk6
android6
react-native6

Programming languages (2)

C++Python

Github contributions (5)

github-logo-circle
ivy-llc/ivy

Nov 2022 - Jan 2023

Convert Machine Learning Code Between Frameworks
Role in this project:
userBack-end Developer
Contributions:24 reviews, 4 commits, 29 PRs in 1 month
Contributions summary:Usama contributed to the ivy project by implementing and testing new functions within the PyTorch frontend. These changes included adding instance methods like `cosh_`, `acosh_`, and `atan2_`, and functional methods like `real` and `fmod`. The user's work involved modifying files in the `ivy/functional/frontends/torch` and `ivy_tests/test_ivy/test_frontends/test_torch` directories. Furthermore, the user also made improvements for the jax and tensorflow backends by adding related logic functions.
pythontensorflowframework-learningtemplatedata-science
usama-baloch/Gaming-Zone

May 2021 - Jun 2024

Contributions:2 pushes, 1 branch in 3 years 1 month
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
Usama Baloch - Machine Learning Engineer at Unify