Code

Improvements on Sparse API.#98

Code

Merged

AndPotap merged 4 commits intomainwilson-labs/cola:mainfrom sparse_improvementswilson-labs/cola:sparse_improvementsCopy head branch name to clipboardAug 29, 2024

+73-28

Commits

Commits on Aug 28, 2024

• Change sparse API to get row and col indices, added transpose dispatch

  AndPotap

  committed
  88d96ab

Commits on Aug 29, 2024

• Fixed JAX operator

  AndPotap

  committed
  931cf00

• Ensured backprop is working

  AndPotap

  committed
  2f95c12

• Merge branch 'main' into sparse_improvements

  AndPotap

  authored
  19a88e9