Log #5. One small neural network, copied into every bot.
october 2026 • ai, machine learning, imitation learning, jax • 2 min read
Up to here every decision came from rules I wrote by hand, and some of them read a map of the whole grid, which no real bot could see. So I replaced the bots' decision-making with a trained model.
52,627
weights in the network, the same copy in every bot
262
numbers each bot sees: its 5×5×5 surroundings, the shape map, a relayed "how far to the work" signal, a compass
19
possible moves to choose from: stay, 6 slides, 12 flips
Moves that aren't legal right now are blocked before the bot chooses, so the network decides which safe move to make, never whether to let go. This is the same idea as Kuan's Neural Particle Automata (one shared network in every particle, seeing only its neighbours), except our output is "which of 19 moves" because the bots have to stay attached.
A few minutes on a CPU. No GPU needed at this size.

Lesson: when a learned policy fails, check whether it's missing information before making it bigger. And test it somewhere that doesn't look like training.