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?