diff --git a/docs/conf.py b/docs/conf.py
index <HASH>..<HASH> 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -96,6 +96,8 @@ exclude_patterns = [
     'notebooks/Neural_Network_and_Data_Loading.ipynb',
     'notebooks/score_matching.ipynb',
     'notebooks/maml.ipynb',
+    # Fails with shape error in XL
+    'notebooks/XLA_in_Python.ipnb'
 ]
 
 # The name of the Pygments (syntax highlighting) style to use.
Temporarily disable XLA_in_Python notebook, pending fixing of bug

Issue: #<I>