Add training code (same as GitHub reasoning-by-superposition-latent)

8f46582

verified

Avra98 commited on 12 days ago