Various aspects relate generally to the beam management procedures in wireless communications systems. Some aspects more specifically relate to the selection of beams for communications to and from a UE and a network entity based on quantized orientation information for a user equipment (UE). In some implementations, a network entity can use the quantized orientation information and a machine learning model to predict a set of beams that may be suitable for communications to and from the UE and the network entity, which may be a subset of the beams that the network entity can generally use for communications to and from the UE and the network entity.