(3) A mechanism to rapidly prototype different matrix structures and compositions of structures. CoLA natively supports PyTorch, JAX, as well as (limited) NumPy if JAX is not installed.