scratchbubbles

scratchbubbles

Scratchubbbles: thoughtbubbles except we can fork into any other token.

Scratchbubbles

Bases: Thoughtbubbles

plot(intermediates: Any) -> Dict[str, Any] staticmethod

intermediates -> [figure]

pca(x: jax.Array, n_components: int) -> jax.Array

x: (N, D) -> (N, n_components). Centered SVD.

vectors_to_colors(embeddings: jax.Array) -> jax.Array

(layers, seq_len, hidden) -> (layers, seq_len, 3) RGB.