Logo

🌱 Getting Started

  • Philosophy
  • Quick Start

🧰 Basic Functionality

  • Walkthrough of Linear Algebra Functionality
  • Implementing new Linear Operators and Dispatch Rules
  • Defining New Dispatch Rules
  • Accessing Lower Level Algorithms (CG, Lanczos, Arnoldi, etc)
  • GPU Support and Changing Operator Device

💡 Example Applications

  • Gaussian Processes from scratch
  • 2nd order optimization of neural nets using Gauss Newton
  • Computing the eigenspectrum of the Hessian of a Neural Network
  • Boundary Value PDEs
  • Diagonalizing a Hamiltonian (PDE eigenvalue problems)
  • Spectral Clustering

🧙‍♂️ Advanced Features

  • jit, vmap, grad, and pytrees

📚 API Reference

  • Linear Algebra
  • Linear Operators
  • Linear Operator Base Class
  • Decompositions
  • Functions
  • Annotations for Linear Operators

🚧 Tricky Bits 🚧

  • Sharp Bits

👩‍💻 Developer Documentation

  • Contributing to CoLA
CoLA
  • Search

© Copyright 2023, Wilson-Labs.

Built with Sphinx using a theme provided by Read the Docs.