Machine Learning Engineer | Open Source Contributor
I am a top contributor to google/flax and actively contribute to CPython and PyTorch.
- AI Security: Understanding and addressing vulnerabilities in machine learning systems
- AI Interpretability: Developing methods to understand how neural networks make decisions
- Large Language Models: Architecture design, training efficiency, and optimization
- AI Agents: Building autonomous systems that can reason and act in complex environments
- Multimodal Systems: Integration of vision, language, and other modalities
primary_frameworks = ["PyTorch", "JAX", "Flax"]
languages = ["Python", "C++", "CUDA"]
areas = ["Deep Learning", "LLMs", "AI Safety", "Interpretability"]- Website: mohsinmehmood.com
- LinkedIn: mohsin-mehmood675
- Location: Pakistan

