• License: Apache-2.0
• Home: https://github.com/wilson-labs/cola
• Development: https://github.com/wilson-labs/cola
• Documentation: https://cola.readthedocs.io/en/latest/
• 9 total downloads
• Last upload: 3 months and 14 days ago

Description

CoLA is a framework for scalable linear algebra, automatically exploiting the structure often found in machine learning problems and beyond. CoLA natively supports PyTorch, Jax, as well as (limited) Numpy if Jax is not installed.