.. _api: ************* API Reference ************* .. toctree:: :maxdepth: 2 api/models api/diagnostics api/samplers api/data_batches api/tensor_utils .. XXX: Add necessary stuff here