Block Matrix Linear Operator - #67

Open

corwinjoy wants to merge 26 commits into

cornellius-gp:mainfrom

corwinjoy:block_tensor_lo

Commits on May 30, 2023

Rebase vs. main
Corwin Joy
committed
e2e32f1
Fix block tensor type signatures
Corwin Joy
committed
15effdf

Commits on May 31, 2023

Get simple test running for BlockTensor
Corwin Joy
committed
4055a7f
Add simple property implementations
Corwin Joy
committed
2cf2db0
Upgrade linear operator to add block / sparse test
Corwin Joy
committed
61d9b8a
Add and document core test cases
Corwin Joy
committed
bbc12ea
Cleanup dead comments
Corwin Joy
committed
bfc843a

Commits on Jun 1, 2023

Update linear_operator/operators/block_tensor_linear_operator.py

corwinjoy
and
dannyfriar
authored
516b369
Update linear_operator/operators/block_tensor_linear_operator.py

corwinjoy
and
dannyfriar
authored
9d12d7c
Improve construction tests and types
Corwin Joy
committed
56809b2
Rename class to MatrixLinearOperator
Corwin Joy
committed
2f9b4b7
Add parts omitted from base test case. Show them as commented out to make it clear what we are skipping.
Corwin Joy
committed
7cbbc54
Rename class to BlockMatrixLinearOperator
Corwin Joy
committed
32e9a52
Fix type signature
Corwin Joy
committed
30ad0ed

Commits on Jun 2, 2023

Improve comments
Corwin Joy
committed
889ce0f

Commits on Jul 12, 2023

Merge branch 'main' into block_tensor_lo
corwinjoy
authored
d15f368

Commits on Jul 18, 2023

Refactor linear_operator_test_case.py into a set of core tests and more advanced tests. This allows us to create and test operators that only support core operations.
Corwin Joy
committed
c557aa3
Merge remote-tracking branch 'origin/block_tensor_lo' into block_tensor_lo
Corwin Joy
committed
d2cb1cc

Commits on Jul 19, 2023

Incorporate review suggestions from Geoff Pleiss.
Corwin Joy
committed
67b8fe9

Commits on Jul 27, 2023

Add comment explaining matmul override.
gpleiss
authored
75b565a
Add jaxtyping requirement for conda
gpleiss
committed
ff0b6a2
Merge branch 'main' into block_tensor_lo
gpleiss
committed
ffc3116
Fix linter
gpleiss
committed
58e8686
Hopefully fix weird CI errors
gpleiss
committed
7f803e3
Refactor BlockMatrixLO._matmul to better adhere to type signatures
gpleiss
committed
c7d094d
BlockMatrixLO takes in a flattened represetation
gpleiss
committed
8d29dd7