Changed
- Make `TreeCRFLayer` zero-initialize transition weights.
- Use vectorization and broadcasting to compute message passing.
- Bump Torch dependency to `v2.0`.
- Swapped last two dimensions of tensors returned from `TreeCRFLayer` for `CrossEntropyLoss` compatibility.
Removed
- Dedicated `TreeMatrix` type for storing a hierarchy.