Notes for learning JAX

As a learning exercise, I recently implemented simple-jax-nn, a simple neural net for the MNIST dataset, written with JAX. This post is my notes on using JAX (as a PyTorch / NumPy user) from working through the JAX docs and a NN tutorial. Simple JAX neural net I managed to get it to use my MacBook’s GPU/Accelerator through the Metal support in jax-metal. It’s a little bit tricky, but will prove useful in future transformer projects that can make use of the hardware. ...

11 Sep 2025 Â· 5 min Â· Gianluca Truda

Streams of beautiful riches

Enter richify.py: a real-time Markdown rendering tool that supports streaming input. Built with Rich and shipped effortlessly with uv.

23 Jan 2025 Â· 1 min Â· Gianluca Truda

DevLog: llmpossible

How I built a command-line LLM on Apple Silicon for RC’s ‘Impossible Stuff Day’

8 Oct 2024 Â· 21 min Â· Gianluca Truda

RC Weeknotes 05

Week 5 of Recurse: Move fast and make games.

16 Sep 2024 Â· 5 min Â· Gianluca Truda

AImong Us: a reverse Turing Test game

Can you outwit a gang of LLMs?

14 Sep 2024 Â· 3 min Â· Gianluca Truda