Beam dependent digital pre-distortion

1. A method for wireless communication at a wireless device, comprising:

applying a first set of digital pre-distortion coefficients to a plurality of antenna elements to form a first transmit beam that is associated with a first loading configuration of the plurality of antenna elements;

determining to switch from using the first transmit beam to using a second transmit beam that is different from the first transmit beam, the second transmit beam associated with a second loading configuration that is different from the first loading configuration;

applying a second set of digital pre-distortion coefficients to the plurality of antenna elements to form the second transmit beam, wherein the second set of digital pre-distortion coefficients is different from the first set of digital pre-distortion coefficients; and

transmitting a signal using the second transmit beam based at least in part on applying the second set of digital pre-distortion coefficients.

2. The method of claim 1 , further comprising:

selecting, from a plurality of digital pre-distortion coefficient sets, the second set of digital pre-distortion coefficients corresponding to the second transmit beam.

3. The method of claim 2 , further comprising:

determining the plurality of digital pre-distortion coefficient sets based at least in part on a calibration process that identifies non-linearity characteristics of the plurality of antenna elements for a plurality of transmit beams including the first transmit beam and the second transmit beam.

4. The method of claim 3 , wherein the calibration process comprises:

performing a digital pre-distortion coefficient calibration for each of the plurality of transmit beams based at least in part on a dynamic calibration schedule that occurs while the wireless device is operating.

5. The method of claim 3 , wherein the calibration process comprises:

performing a digital pre-distortion coefficient calibration for each of the plurality of transmit beams based at least in part on powering on the wireless device.

6. The method of claim 1 , wherein:

the plurality of antenna elements are grouped into a plurality of radio frequency modules; and

each antenna element of the plurality of antenna elements is associated with a power amplifier.

7. The method of claim 6 , wherein the first transmit beam is associated with a first direction and the second transmit beam is associated with a second direction that is different from the first direction.

8. The method of claim 6 , wherein the first set of digital pre-distortion coefficients and the second set of digital pre-distortion coefficients are applied by a single digital pre-distortion engine that is common to the plurality of radio frequency modules.

9. The method of claim 1 , wherein the wireless device is a base station.

10. The method of claim 1 , wherein the wireless device is a user equipment.

11. An apparatus for wireless communication at a wireless device, comprising:

a processor;

memory coupled with the processor; and

instructions stored in the memory and executable by the processor to cause the apparatus to:

apply a first set of digital pre-distortion coefficients to a plurality of antenna elements to form a first transmit beam that is associated with a first loading configuration of the plurality of antenna elements;

determine to switch from using the first transmit beam to using a second transmit beam that is different from the first transmit beam, the second transmit beam associated with a second loading configuration that is different from the first loading configuration;

apply a second set of digital pre-distortion coefficients to the plurality of antenna elements to form the second transmit beam, wherein the second set of digital pre-distortion coefficients is different from the first set of digital pre-distortion coefficients; and

transmit a signal using the second transmit beam based at least in part on applying the second set of digital pre-distortion coefficients.

12. The apparatus of claim 11 , wherein the instructions are further executable by the processor to cause the apparatus to:

select, from a plurality of digital pre-distortion coefficient sets, the second set of digital pre-distortion coefficients corresponding to the second transmit beam.

13. The apparatus of claim 12 , wherein the instructions are further executable by the processor to cause the apparatus to:

determine the plurality of digital pre-distortion coefficient sets based at least in part on a calibration process that identifies non-linearity characteristics of the plurality of antenna elements for a plurality of transmit beams including the first transmit beam and the second transmit beam.

14. The apparatus of claim 13 , wherein the calibration process comprises instructions that are executable by the processor to cause the apparatus to:

perform a digital pre-distortion coefficient calibration for each of the plurality of transmit beams based at least in part on a dynamic calibration schedule that occurs while the wireless device is operating.

15. The apparatus of claim 13 , wherein the calibration process comprises instructions that are executable by the processor to cause the apparatus to:

perform a digital pre-distortion coefficient calibration for each of the plurality of transmit beams based at least in part on powering on the wireless device.

16. The apparatus of claim 11 , wherein:

the plurality of antenna elements are grouped into a plurality of radio frequency modules; and

each antenna element of the plurality of antenna elements is associated with a power amplifier.

17. The apparatus of claim 16 , wherein the first transmit beam is associated with a first direction and the second transmit beam is associated with a second direction that is different from the first direction.

18. The apparatus of claim 16 , wherein the first set of digital pre-distortion coefficients and the second set of digital pre-distortion coefficients are applied by a single digital pre-distortion engine that is common to the plurality of radio frequency modules.

19. The apparatus of claim 11 , wherein the wireless device is a base station.

20. The apparatus of claim 11 , wherein the wireless device is a user equipment.

21. An apparatus for wireless communication at a wireless device, comprising:

means for applying a first set of digital pre-distortion coefficients to a plurality of antenna elements to form a first transmit beam that is associated with a first loading configuration of the plurality of antenna elements;

means for determining to switch from using the first transmit beam to using a second transmit beam that is different from the first transmit beam, the second transmit beam associated with a second loading configuration that is different from the first loading configuration;

means for applying a second set of digital pre-distortion coefficients to the plurality of antenna elements to form the second transmit beam, wherein the second set of digital pre-distortion coefficients is different from the first set of digital pre-distortion coefficients; and

means for transmitting a signal using the second transmit beam based at least in part on applying the second set of digital pre-distortion coefficients.

22. The apparatus of claim 21 , further comprising:

means for selecting, from a plurality of digital pre-distortion coefficient sets, the second set of digital pre-distortion coefficients corresponding to the second transmit beam.

23. The apparatus of claim 22 , further comprising:

means for determining the plurality of digital pre-distortion coefficient sets based at least in part on a calibration process that identifies non-linearity characteristics of the plurality of antenna elements for a plurality of transmit beams including the first transmit beam and the second transmit beam.

24. The apparatus of claim 23 , further comprising:

means for performing a digital pre-distortion coefficient calibration for each of the plurality of transmit beams based at least in part on a dynamic calibration schedule that occurs while the wireless device is operating.

25. The apparatus of claim 23 , further comprising:

means for performing a digital pre-distortion coefficient calibration for each of the plurality of transmit beams based at least in part on powering on the wireless device.

26. The apparatus of claim 21 , wherein:

the plurality of antenna elements are grouped into a plurality of radio frequency modules; and

each antenna element of the plurality of antenna elements is associated with a power amplifier.

27. The apparatus of claim 26 , wherein the first transmit beam is associated with a first direction and the second transmit beam is associated with a second direction that is different from the first direction.

28. The apparatus of claim 26 , wherein the first set of digital pre-distortion coefficients and the second set of digital pre-distortion coefficients are applied by a single digital pre-distortion engine that is common to the plurality of radio frequency modules.

29. A non-transitory computer-readable medium storing code for wireless communication at a wireless device, the code comprising instructions executable by a processor to:

apply a first set of digital pre-distortion coefficients to a plurality of antenna elements to form a first transmit beam that is associated with a first loading configuration of the plurality of antenna elements;

determine to switch from using the first transmit beam to using a second transmit beam that is different from the first transmit beam, the second transmit beam associated with a second loading configuration that is different from the first loading configuration;

apply a second set of digital pre-distortion coefficients to the plurality of antenna elements to form the second transmit beam, wherein the second set of digital pre-distortion coefficients is different from the first set of digital pre-distortion coefficients; and

transmit a signal using the second transmit beam based at least in part on applying the second set of digital pre-distortion coefficients.

30. The non-transitory computer-readable medium of claim 29 , wherein the instructions are further executable by the processor to:

select, from a plurality of digital pre-distortion coefficient sets, the second set of digital pre-distortion coefficients corresponding to the second transmit beam.