--- a
+++ b/doc/source/api/utils.rst
@@ -0,0 +1,98 @@
+torchdrug.utils
+===============
+
+Visualization
+-------------
+
+.. currentmodule:: torchdrug.utils.plot
+
+.. autofunction:: reaction
+
+.. autofunction:: highlight
+
+.. autofunction:: echarts
+
+Auxiliary Torch Functions
+-------------------------
+
+.. currentmodule:: torchdrug.utils
+
+.. autofunction:: load_extension
+
+.. autofunction:: cpu
+
+.. autofunction:: cuda
+
+.. autofunction:: detach
+
+.. autofunction:: clone
+
+.. autofunction:: mean
+
+.. autofunction:: cat
+
+.. autofunction:: stack
+
+.. autofunction:: sparse_coo_tensor
+
+Distributed Communication
+-------------------------
+
+.. currentmodule:: torchdrug.utils.comm
+
+.. autofunction:: init_process_group
+
+.. autofunction:: get_cpu_count
+
+.. autofunction:: get_group
+
+.. autofunction:: get_rank
+
+.. autofunction:: get_world_size
+
+.. autofunction:: reduce
+
+.. autofunction:: stack
+
+.. autofunction:: cat
+
+File Processing
+---------------
+
+.. currentmodule:: torchdrug.utils
+
+.. autofunction:: download
+
+.. autofunction:: extract
+
+.. autofunction:: compute_md5
+
+.. autofunction:: get_line_count
+
+Commandline I/O
+---------------
+
+.. autofunction:: input_choice
+
+.. autofunction:: literal_eval
+
+.. autofunction:: no_rdkit_log
+
+.. autofunction:: capture_rdkit_log
+
+.. automodule:: torchdrug.utils.pretty
+    :members:
+
+Decorator
+---------
+
+.. currentmodule:: torchdrug.utils
+
+.. autofunction:: cached_property
+
+Helper functions
+----------------
+
+.. currentmodule:: torchdrug.utils
+
+.. autofunction:: copy_args