diff --git a/mjx/mujoco/mjx/benchmark/benchmark.py b/mjx/mujoco/mjx/benchmark/benchmark.py
index b0a59af0..61112953 100644
--- a/mjx/mujoco/mjx/benchmark/benchmark.py
+++ b/mjx/mujoco/mjx/benchmark/benchmark.py
@@ -38,9 +38,9 @@ _BATCH_SIZE = {
     ('humanoid', 'TPU v5 lite'): 1024,
     ('barkour', 'TPU v5 lite'): 1024,
     ('shadow_hand', 'TPU v5 lite'): 1024,
-    ('humanoid', 'Tesla V100-SXM2-16GB'): 8192,
-    ('barkour', 'Tesla V100-SXM2-16GB'): 8192,
-    ('shadow_hand', 'Tesla V100-SXM2-16GB'): 4096,
+    ('humanoid', 'NVIDIA A100-SXM4-40GB'): 8192,
+    ('barkour', 'NVIDIA A100-SXM4-40GB'): 8192,
+    ('shadow_hand', 'NVIDIA A100-SXM4-40GB'): 4096,
     ('humanoid', 'cpu'): 64,
     ('barkour', 'cpu'): 64,
     ('shadow_hand', 'cpu'): 64,