YUN, Junwoo
YUN, Junwoo
## Proposed changes Added commonly used losses with tests ## Checklist Put an `x` in the boxes that apply. - [x] I have read the [CONTRIBUTING](https://github.com/ml-explore/mlx/blob/main/CONTRIBUTING.md) document - [x] I...
Yes, but still there's problem since you are concatenating not doing +. ```python class TransformerLM(nn.Module): def __init__(self, vocab_size: int, num_layers: int, dims: int, num_heads: int): super().__init__() self.my_encoder = transformer.MyPosEncoding(dims) self.your_encoder...