Solved by keyboardAnt in post #2 
              
              
                Sampling points on the unit sphere (i.e. points of norm 1) is possible by using BoTorch’s sample_hypersphere. Referece: https://github.com/pytorch/botorch/pull/369/files