Matryoshka2dLoss

This release focuses on correctness and performance, rebuilding gradient-cached losses on a shared engine for faster training and fixing several gradient bugs. It also significantly improves inference speed when using `model.compile()`.