For optimizers.py, are you relying purely on boolean arithmetic for these masks, or did you find a more elegant mathematical workaround to avoid the branching overhead entirely?

Thanks to that, I discovered additional errors and corrected them in various ways. Thank you.

i naver imagined it i've made the corrections thank you!