braket.annealing package