How to use the WM8904 codec in Jetson AGX Xavier?

Hi,
A user has enabled it on TX2 NX. You may refer to this topic and give it a try:
WM8904 audio driver support for Jetson TX2 NX

Besides, FE-PI audio v2 board is supported in 40-pin expansion header. You can enable it through jetson-io.py and refer to the device tree. To check which part has to be modified in device tree.