Brevitas: quantization-aware training in PyTorch - View it on GitHub