fairseq2.device

The device module provides device management utilities for working with CPUs, GPUs, and other computing devices in a unified way.

Coming soon: This documentation is being developed. The device module includes:

• Device abstraction and management

• Device detection and selection

• Cross-device tensor operations

Please refer to the source code and examples in the meantime.