CoolFace
Apppublic

PRATYUSH-BHARDWAJ/Cortex-A-0.5

sourceHugging Faceapache-2.0updated 3mo agoView on Hugging Face
0likes
App README

๐Ÿง  Cortex-A 0.5 โ€” live latent-AR playground

This Space serves the latest weights of Cortex-A 0.5, a ~0.73B two-level language model pre-trained from scratch in JAX/Flax on a single TPU v5e-8.

  • โ€”A deep latent-AR planner (32 layers over 5-token chunks) predicts the next chunk embedding through a KV-cached decode.
  • โ€”A shallow AR writer decodes each token conditioned on the full planner plan (cond_fuse(concat(token, plan))), the planner advancing once per 5-token chunk.
  • โ€”A tied/factorized head reads the tokens out โ€” standard next-token decoding.

The latent backbone is always on: its plan conditions every generated token, so what you see is the real two-level pipeline, not the writer alone. It's a base model โ€” type the start of a sentence and it continues it.