"""Cortex-A: an efficiency-focused ~0.55B small language model in JAX/Flax."""