diff --git a/docs/_static/style.css b/docs/_static/style.css
index 36b54b843..51ab72d71 100644
--- a/docs/_static/style.css
+++ b/docs/_static/style.css
@@ -296,3 +296,13 @@ html[data-theme="light"] .highlight span.gt {
 html[data-theme="light"] .highlight span.gr {
   color: #ff0000;
 }
+
+.header-article-items__start {
+  display: flex;
+  flex-direction: row;
+  gap: 0.5em;
+}
+
+.bd-breadcrumbs {
+  margin-bottom: 0;
+}
diff --git a/docs/conf.py b/docs/conf.py
index 285a5abc9..bbc417af1 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -172,6 +172,7 @@ html_theme_options = {
     'repository_url': 'https://github.com/jax-ml/jax',
     'use_repository_button': True,     # add a "link to repository" button
     'navigation_with_keys': False,
+    'article_header_start': ['toggle-primary-sidebar.html', 'breadcrumbs'],
 }
 
 # The name of an image file (relative to this directory) to place at the top