Fix docs for TransformerBlock and remove incompatible functions - #37

Open

mashu wants to merge 7 commits into

MurrellGroup:mainfrom

Commits on Oct 31, 2025

src/transformers/block.jl: Update the docstrings to explain forward pass of the TransformerBlock.
mashu
committed
22fbe4f
src/Utils/masks.jl: Remove confusing log-space self_att_padding_mask and cross_att_padding_mask which are not compatible with new apply_pad_mask.
mashu
committed
b1f7e04
test/UNet/test_unet.jl: Simplify tests, use smaller tensors, bump gradient tests and ignore UNet for codecov reporting.
mashu
committed
7a9cd1b