Merge pull request #44 from gomlx/fused-ops-refactor

Updated to renaming of `compute.AxesLayout` -> `compute.AttentionAxesLayout`

Merge pull request #44 from gomlx/fused-ops-refactor

Updated to renaming of `compute.AxesLayout` -> `compute.AttentionAxesLayout`

Updated CHANGELOG.

Updated internal module dependency.

Updated internal module dependency.

Merge pull request #29 from gomlx/fixed-nvidia-cuda-path

- Fixed CUDA library paths -- it was hardcoded to a test directory.

Updated internal module dependency.

Merge pull request #28 from gomlx/bump-v0.2.0

Updated docs and comments to v0.2.0 bump.

Updated pjrt_installer internal dependency.

Added badges for tests on other platforms.