Stories by jxmorris12

It's time to become an ML engineer (2022)

Approximating KL Divergence (2020)

The Ultra-Scale Playbook: Training LLMs on GPU Clusters

Implementing LLaMA3 in 100 Lines of Pure Jax

Outperforming cuBLAS on H100: A Worklog

Gravel Map

Stochastic Integration for Poets

NASA writes space-proof code [video]

Large Lambda Model

Softmax forever, or why I like softmax

Diffusion Without Tears

AGI Safety Course Workbook

AI Nationalism – Ian Hogarth

A Beginners' Guide to Misprints in Magic

Flow with What You Know

Going with the Flow: An Introduction to Normalizing Flows

Diffusion Meets Flow Matching: Two Sides of the Same Coin

Muon: An optimizer for hidden layers in neural networks

1
2
3
4
5
6
7
8