XOR через ReLU-блоки

Реализуйте побитовый XOR двух неотрицательных чисел через простые ReLU-блоки.

Для каждого бита a_i и b_i значение XOR можно записать как:

relu(a_i - b_i) + relu(b_i - a_i)

где relu(x) = max(0, x).

Сигнатура

def xor_relu(a: int, b: int, bits: int = 16) -> int:

Обработайте младшие bits битов и верните целое число.