Ali Nawaz – Medium

Ali Nawaz

Ali Nawaz

·

May 6

Making Financial Rules Portable (Without Losing Your Mind)

Author: Ali Nawaz, FINOS Ambassador

Making Financial Rules Portable (Without Losing Your Mind)

Ali Nawaz

·

Jan 26

jit, grad, and vmap in JAX — One simple idea behind all three: transforming functions

If you’re new to JAX, you’ll keep seeing three words everywhere:

jit, grad, and vmap in JAX — One simple idea behind all three: transforming functions

Ali Nawaz

·

Dec 27, 2025

What the Compact API in Flax Really Does: The Design Choice That Keeps Parameter Creation…

Why this topic matters

What the Compact API in Flax Really Does: The Design Choice That Keeps Parameter Creation…

Ali Nawaz

·

Dec 25, 2025

How Flax Modules Work Internally: The Design Choice That Keeps JAX Functional

Why this topic matters

How Flax Modules Work Internally: The Design Choice That Keeps JAX Functional

Ali Nawaz

·

Dec 7, 2025

How JAX Compiles Your Code: The Secret Relationship Between JAX and XLA

In the previous post, we explored why JAX thinks differently, why it separates parameters from computation, and how its functional mindset…

How JAX Compiles Your Code: The Secret Relationship Between JAX and XLA

Ali Nawaz

·

Dec 5, 2025

Why JAX Thinks Differently: The Functional Mindset Behind its Power

In the previous post, we explored how JAX compares to TensorFlow and PyTorch, and why its speed-focused design feels so different. But one…

Why JAX Thinks Differently: The Functional Mindset Behind its Power

Ali Nawaz

·

Nov 2, 2025

JAX vs. TensorFlow vs. PyTorch: A Deeper Look for Beginners

In the previous post, we walked through “Why JAX Exists? The Need for Speed and Simplicity in ML”

JAX vs. TensorFlow vs. PyTorch: A Deeper Look for Beginners

Ali Nawaz

·

Nov 1, 2025

Why JAX Exists? The Need for Speed and Simplicity in ML

In the previous post, we walked through “Your First Guide to JAX: NumPy with Superpowers” and got comfortable with JAX basics.

Why JAX Exists? The Need for Speed and Simplicity in ML

Ali Nawaz

·

Oct 26, 2025

Your First Guide to JAX: NumPy with Superpowers

Have you ever used NumPy? It’s that amazing Python library everyone uses for working with lists or “arrays” of numbers. It’s fast…

Your First Guide to JAX: NumPy with Superpowers

Ali Nawaz

·

Oct 10, 2025

What Is JAX? A Friendly Introduction for ML Beginners

Have you ever used TensorFlow or PyTorch and wished your code ran much faster, especially when dealing with massive datasets or complex new…

What Is JAX? A Friendly Introduction for ML Beginners

Ali Nawaz

Ali Nawaz

Grandmaster ML Campaigns @Google | Community Organizer @TensorFlow User Group Islamabad & Machine Learning Pakistan | DevOps Institute Ambassador @PeopleCert

Following

Help

Status

About

Careers

Press

Blog

Privacy

Rules

Terms

Text to speech