HomeAI & Machine LearningMachine Learning Optimization Simulator

Machine Learning Optimization Simulator

Interactive 3D loss-surface simulator: watch SGD, Momentum and Adam race a ball down a bumpy loss landscape at the same time, tuning learning rate and momentum to see convergence speed and local-minimum traps in real time.

AI & Machine Learning3DModerate60 FPS
machine-learning-optimization ↗ Open standalone

Training a machine learning model means searching a high-dimensional loss surface for a point where the loss is as low as possible. This simulator visualizes that search directly in 3D: a bumpy landscape stands in for a model's loss function, with a deep global minimum and two shallower local-minimum traps carved into it. Three balls — one per optimizer — start from the same point and race downhill using SGD, Momentum and Adam's real update rules. Tune the learning rate and momentum, toggle gradient noise to mimic mini-batch stochasticity, and watch which optimizer converges fastest, which overshoots, and which gets stuck in a shallow trap instead of reaching the true minimum.

⚙ Under the hood

Interactive 3D loss-surface simulator: watch SGD, Momentum and Adam race a ball down a bumpy loss landscape at the same time, tuning learning rate and momentum to see convergence speed and local-minimum traps in real time.

Three.jsmachine-learninggradient-descentoptimizationneural-networksadam-optimizer

3D · Three.js / WebGL renderer · 60 FPS target · runs fully client-side, no install

What did you find?

Add reproduction steps (optional)