CUTLASS 3.1 Python interface documentation (#917)

* Add 12.1 Dockerfile

* Add 3.1 docs

python/docs/_static/tabs.js

@@ -0,0 +1,21 @@
+var labels_by_text = {};
+
+function ready() {
+  var li = document.getElementsByClassName("tab-label");
+  for (const label of li) {
+    label.onclick = onLabelClick;
+    const text = label.textContent;
+    if (!labels_by_text[text]) {
+      labels_by_text[text] = [];
+    }
+    labels_by_text[text].push(label);
+  }
+}
+
+function onLabelClick() {
+  // Activate other labels with the same text.
+  for (label of labels_by_text[this.textContent]) {
+    label.previousSibling.checked = true;
+  }
+}
+document.addEventListener("DOMContentLoaded", ready, false);