Neural network polytopes (Colab notebook) — LessWrong