Wesley Kambale

@kambale

Machine Learning Engineer | AI Community Leader | Technical Educator

Mbarara, UgandaJoined August 2021

About

I'm a Machine Learning Engineer passionate about building production-ready ML systems for the African market. With experience in TensorFlow, Keras, and Python-based workflows, I help teams bridge the gap between machine learning research and real-world deployment—especially on resource-constrained devices. I'm also a Google Developer Expert in AI. I regularly speak at tech conferences including PyCon Africa, DevFest Kampala, DevFest Nairobi and more and also write technical articles on AI/ML here.

Available for

Nothing here yet.

Wesley Kambale's blogs

Wesley Kambalekambale.dev34 posts

Articles Comments

Recently published

WKWesley Kambalekambale.devMar 4 · 10 min read

Custom training loop from scratch in JAX

For the past three weeks, we've been building up to this moment. Week 1 taught us that JAX is fast. Week 2 showed us how to eliminate loops with vmap and compute gradients with grad. Week 3 gave us Fl

0

WKWesley Kambalekambale.devFeb 23 · 11 min read

Building Neural Networks with Flax NNX

Over the past two weeks, we've learned that JAX is fast (jit), that it eliminates loops (vmap), and that it computes gradients automatically (grad). These are powerful primitives. But if you've been f

0

WKWesley Kambalekambale.devFeb 9 · 10 min read

Transformations That Change Everything

Last week, we learned that JAX makes code fast through JIT compilation. We took a matrix multiplication from 2 seconds to 0.001 seconds with a single decorator. But speed isn't JAX's only trick. The real power of JAX lies in its transformations; func...

0

WKWesley Kambalekambale.devFeb 2 · 9 min read

Why JAX? The NumPy You Know, But Faster

If you've been doing machine learning in Python for any length of time, you've written code like this: import numpy as np x = np.random.randn(1000, 1000) y = np.random.randn(1000, 1000) result = np.dot(x, y) NumPy is comfortable. It's the first thi...

0

WKWesley Kambalekambale.devDec 10, 2025 · 4 min read

The reverse turing test: We must now prove we are “dumb” to beat AI

In the dystopian logic of the digital age, a new anxiety has gripped the writing world. From university lecture halls in Makerere to newsrooms in Kampala, humans are facing a pressure that would have seemed laughable just two years ago: the pressure ...

0

Wesley Kambale

About

Available for

Wesley Kambale's blogs

Recently published

Custom training loop from scratch in JAX

Building Neural Networks with Flax NNX

Transformations That Change Everything

Why JAX? The NumPy You Know, But Faster

The reverse turing test: We must now prove we are “dumb” to beat AI

Search Hashnode

Wesley Kambale

About

Available for

Wesley Kambale's blogs

Recently published

Custom training loop from scratch in JAX

Building Neural Networks with Flax NNX

Transformations That Change Everything

Why JAX? The NumPy You Know, But Faster

The reverse turing test: We must now prove we are “dumb” to beat AI