# kd.inspect

[[[Source]]](https://github.com/google-research/kauldron/tree/main/kauldron/inspect/__init__.py)

```{eval-rst}
.. automodule:: kauldron.inspect
  :no-members:
```

## Symbols


### Class

|  |  |
--- | ---
[kd.inspect.Profiler](Profiler) | `kd.inspect.Profiler`.

### Function

|  |  |
--- | ---
[kd.inspect.get_batch_stats](get_batch_stats) | Return `pd.DataFrame` containing the batch stats.
[kd.inspect.get_colab_model_overview](get_colab_model_overview) | Return `pd.DataFrame` for displaying the model params, inputs,...
[kd.inspect.get_connection_graph](get_connection_graph) | Build the graphviz.
[kd.inspect.json_spec_like](json_spec_like) | Convert `etree.spec_like` output to json and displays it in colab form.
[kd.inspect.lower_trainstep](lower_trainstep) | Returns lowered trainerstep.step.
[kd.inspect.plot_batch](plot_batch) | Display batch images.
[kd.inspect.plot_context](plot_context) | Display the context structure.
[kd.inspect.plot_schedules](plot_schedules) | Overview plot for (nested) dict of schedules.
[kd.inspect.plot_sharding](plot_sharding) | Plot sharding.
[kd.inspect.show_trainer_info](show_trainer_info) | Display various plot on the trainer.

```{toctree}
:hidden:

Profiler
get_batch_stats
get_colab_model_overview
get_connection_graph
json_spec_like
lower_trainstep
plot_batch
plot_context
plot_schedules
plot_sharding
show_trainer_info
```