Add block tri diagonal class to Torax to represent the discrete system#2116

copybara-service[bot] merged 1 commit into

maingoogle-deepmind/torax:mainfrom

test_884493417google-deepmind/torax:test_884493417Copy head branch name to clipboard