diff --git a/dev/bench/data.js b/dev/bench/data.js
index bfa0f0e07..ffa6552bf 100644
--- a/dev/bench/data.js
+++ b/dev/bench/data.js
@@ -1,5 +1,5 @@
 window.BENCHMARK_DATA = {
-  "lastUpdate": 1699649410174,
+  "lastUpdate": 1699893972326,
   "repoUrl": "https://github.com/pytorch/tensordict",
   "entries": {
     "CPU Benchmark Results": [
@@ -132751,6 +132751,765 @@ window.BENCHMARK_DATA = {
             "extra": "mean: 549.7634749288472 usec\nrounds: 1735"
           }
         ]
+      },
+      {
+        "commit": {
+          "author": {
+            "email": "vmoens@meta.com",
+            "name": "Vincent Moens",
+            "username": "vmoens"
+          },
+          "committer": {
+            "email": "noreply@github.com",
+            "name": "GitHub",
+            "username": "web-flow"
+          },
+          "distinct": true,
+          "id": "95bd846a859b201377a6baaefaf11bb97a60b225",
+          "message": "[Refactor] Refactor implement_for (#556)",
+          "timestamp": "2023-11-13T16:41:57Z",
+          "tree_id": "d04abaeb99bb50005ac2c5562012696f71addc86",
+          "url": "https://github.com/pytorch/tensordict/commit/95bd846a859b201377a6baaefaf11bb97a60b225"
+        },
+        "date": 1699893968699,
+        "tool": "pytest",
+        "benches": [
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_plain_set_nested",
+            "value": 66530.68379167035,
+            "unit": "iter/sec",
+            "range": "stddev: 7.753719710162576e-7",
+            "extra": "mean: 15.03065868271145 usec\nrounds: 7181"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_plain_set_stack_nested",
+            "value": 7082.7661712970785,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000013780691515867099",
+            "extra": "mean: 141.18777548417475 usec\nrounds: 775"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_plain_set_nested_inplace",
+            "value": 56167.3438846038,
+            "unit": "iter/sec",
+            "range": "stddev: 0.0000019801067001355906",
+            "extra": "mean: 17.80393963536013 usec\nrounds: 14313"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_plain_set_stack_nested_inplace",
+            "value": 6018.221341741382,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000004158232309490857",
+            "extra": "mean: 166.16205074814485 usec\nrounds: 4946"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_items",
+            "value": 417083.0227142244,
+            "unit": "iter/sec",
+            "range": "stddev: 1.7978361478434972e-7",
+            "extra": "mean: 2.397604183196823 usec\nrounds: 130141"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_items_nested",
+            "value": 3740.0903075638275,
+            "unit": "iter/sec",
+            "range": "stddev: 0.0000055182927085688135",
+            "extra": "mean: 267.3732230416028 usec\nrounds: 3255"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_items_nested_locked",
+            "value": 3727.395573756905,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000004517347639408502",
+            "extra": "mean: 268.28384060994176 usec\nrounds: 3344"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_items_nested_leaf",
+            "value": 6062.691130594969,
+            "unit": "iter/sec",
+            "range": "stddev: 0.0000075706434118443275",
+            "extra": "mean: 164.94325349242456 usec\nrounds: 5369"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_items_stack_nested",
+            "value": 729.8556207808384,
+            "unit": "iter/sec",
+            "range": "stddev: 0.0000357296952740909",
+            "extra": "mean: 1.370133998461431 msec\nrounds: 650"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_items_stack_nested_leaf",
+            "value": 809.5452151613761,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000017882458057630636",
+            "extra": "mean: 1.2352614545447698 msec\nrounds: 682"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_items_stack_nested_locked",
+            "value": 1335.146748483096,
+            "unit": "iter/sec",
+            "range": "stddev: 0.00003109397168367489",
+            "extra": "mean: 748.9813394191558 usec\nrounds: 1205"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_keys",
+            "value": 258655.04429921834,
+            "unit": "iter/sec",
+            "range": "stddev: 2.3390132204647604e-7",
+            "extra": "mean: 3.866153094788192 usec\nrounds: 87018"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_keys_nested",
+            "value": 6850.158474831004,
+            "unit": "iter/sec",
+            "range": "stddev: 0.0005626203358268857",
+            "extra": "mean: 145.98202416399866 usec\nrounds: 5711"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_keys_nested_locked",
+            "value": 7097.8733943224715,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000006272905259087805",
+            "extra": "mean: 140.8872692488276 usec\nrounds: 6455"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_keys_nested_leaf",
+            "value": 7228.0288756756645,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000005444356986180705",
+            "extra": "mean: 138.3503050693778 usec\nrounds: 6569"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_keys_stack_nested",
+            "value": 784.1324484095323,
+            "unit": "iter/sec",
+            "range": "stddev: 0.00006416480536972277",
+            "extra": "mean: 1.2752947566808581 msec\nrounds: 711"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_keys_stack_nested_leaf",
+            "value": 783.2696327841103,
+            "unit": "iter/sec",
+            "range": "stddev: 0.00001816676837722952",
+            "extra": "mean: 1.2766995656981206 msec\nrounds: 723"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_keys_stack_nested_locked",
+            "value": 1593.3293226728294,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000014212475703126074",
+            "extra": "mean: 627.6166425673305 usec\nrounds: 1480"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_values",
+            "value": 845772.0387168889,
+            "unit": "iter/sec",
+            "range": "stddev: 7.01141685151914e-8",
+            "extra": "mean: 1.182351690790214 usec\nrounds: 194553"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_values_nested",
+            "value": 21332.37138805239,
+            "unit": "iter/sec",
+            "range": "stddev: 0.0000015309863414778178",
+            "extra": "mean: 46.8771137446102 usec\nrounds: 16581"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_values_nested_locked",
+            "value": 21221.31707571215,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000002236478318868103",
+            "extra": "mean: 47.12242866134366 usec\nrounds: 19092"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_values_nested_leaf",
+            "value": 24095.079424428623,
+            "unit": "iter/sec",
+            "range": "stddev: 0.0000011888070306048868",
+            "extra": "mean: 41.50224958321395 usec\nrounds: 19797"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_values_stack_nested",
+            "value": 913.3399020012109,
+            "unit": "iter/sec",
+            "range": "stddev: 0.00006393118796787576",
+            "extra": "mean: 1.0948826365834985 msec\nrounds: 831"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_values_stack_nested_leaf",
+            "value": 930.0471817219013,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000018262928677597814",
+            "extra": "mean: 1.0752142683219437 msec\nrounds: 846"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_values_stack_nested_locked",
+            "value": 2050.1678769335654,
+            "unit": "iter/sec",
+            "range": "stddev: 0.00002464617373412671",
+            "extra": "mean: 487.76493439927435 usec\nrounds: 1875"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_membership",
+            "value": 760926.1655085982,
+            "unit": "iter/sec",
+            "range": "stddev: 1.363045183012674e-7",
+            "extra": "mean: 1.3141879532183078 usec\nrounds: 125079"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_membership_nested",
+            "value": 369840.56838582305,
+            "unit": "iter/sec",
+            "range": "stddev: 2.1686074412475223e-7",
+            "extra": "mean: 2.703867789205822 usec\nrounds: 84161"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_membership_nested_leaf",
+            "value": 363714.735014724,
+            "unit": "iter/sec",
+            "range": "stddev: 2.2487314406482721e-7",
+            "extra": "mean: 2.749407444159549 usec\nrounds: 111149"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_membership_stacked_nested",
+            "value": 89625.96782661686,
+            "unit": "iter/sec",
+            "range": "stddev: 5.927608046912534e-7",
+            "extra": "mean: 11.15748063032936 usec\nrounds: 37120"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_membership_stacked_nested_leaf",
+            "value": 89427.82665514693,
+            "unit": "iter/sec",
+            "range": "stddev: 0.0000010754015251001168",
+            "extra": "mean: 11.182201753109988 usec\nrounds: 50770"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_membership_nested_last",
+            "value": 175662.21595489644,
+            "unit": "iter/sec",
+            "range": "stddev: 3.4522104613743385e-7",
+            "extra": "mean: 5.692743852535498 usec\nrounds: 59130"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_membership_nested_leaf_last",
+            "value": 175719.50693528447,
+            "unit": "iter/sec",
+            "range": "stddev: 3.680974647052708e-7",
+            "extra": "mean: 5.690887810015816 usec\nrounds: 77137"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_membership_stacked_nested_last",
+            "value": 5710.004675276012,
+            "unit": "iter/sec",
+            "range": "stddev: 0.00000387307231991841",
+            "extra": "mean: 175.13120511616074 usec\nrounds: 4300"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_membership_stacked_nested_leaf_last",
+            "value": 76089.99127840978,
+            "unit": "iter/sec",
+            "range": "stddev: 6.511883360236656e-7",
+            "extra": "mean: 13.142332956000034 usec\nrounds: 32704"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_nested_getleaf",
+            "value": 84329.68328434027,
+            "unit": "iter/sec",
+            "range": "stddev: 8.258349483853592e-7",
+            "extra": "mean: 11.858220748063648 usec\nrounds: 54723"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_nested_get",
+            "value": 87650.2225795256,
+            "unit": "iter/sec",
+            "range": "stddev: 0.0000015630809207627629",
+            "extra": "mean: 11.408984148245532 usec\nrounds: 60750"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_stacked_getleaf",
+            "value": 1778.1273283269015,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000011303376420200226",
+            "extra": "mean: 562.3894217637006 usec\nrounds: 1553"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_stacked_get",
+            "value": 1869.3921867974648,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000009193486180499061",
+            "extra": "mean: 534.9332296681641 usec\nrounds: 1746"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_nested_getitemleaf",
+            "value": 84144.52032680689,
+            "unit": "iter/sec",
+            "range": "stddev: 7.531922065170434e-7",
+            "extra": "mean: 11.884315177222758 usec\nrounds: 54160"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_nested_getitem",
+            "value": 88151.51951028492,
+            "unit": "iter/sec",
+            "range": "stddev: 5.157613495872549e-7",
+            "extra": "mean: 11.344103942341308 usec\nrounds: 59841"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_stacked_getitemleaf",
+            "value": 1776.8454325828034,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000014217564379293843",
+            "extra": "mean: 562.7951546389777 usec\nrounds: 1552"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_stacked_getitem",
+            "value": 1864.647632489831,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000029756521263878457",
+            "extra": "mean: 536.2943553387177 usec\nrounds: 1742"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_lock_nested",
+            "value": 1144.9887913890084,
+            "unit": "iter/sec",
+            "range": "stddev: 0.00007162517203854985",
+            "extra": "mean: 873.3709950006414 usec\nrounds: 1000"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_lock_stack_nested",
+            "value": 78.07157710802888,
+            "unit": "iter/sec",
+            "range": "stddev: 0.013896376259101376",
+            "extra": "mean: 12.808758795999267 msec\nrounds: 1000"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_unlock_nested",
+            "value": 1073.9517089690398,
+            "unit": "iter/sec",
+            "range": "stddev: 0.0016149582741539366",
+            "extra": "mean: 931.1405640016801 usec\nrounds: 1000"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_unlock_stack_nested",
+            "value": 75.78034370349893,
+            "unit": "iter/sec",
+            "range": "stddev: 0.013952766030201772",
+            "extra": "mean: 13.19603410500008 msec\nrounds: 1000"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_flatten_speed",
+            "value": 1479.7286456636443,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000055947663721404834",
+            "extra": "mean: 675.7995818561108 usec\nrounds: 1411"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_unflatten_speed",
+            "value": 852.958916376577,
+            "unit": "iter/sec",
+            "range": "stddev: 0.00006537300668330612",
+            "extra": "mean: 1.1723894091500477 msec\nrounds: 765"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_common_ops",
+            "value": 1563.4570002345467,
+            "unit": "iter/sec",
+            "range": "stddev: 0.00013773484589784882",
+            "extra": "mean: 639.6082526414107 usec\nrounds: 1041"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_creation",
+            "value": 454775.3355954383,
+            "unit": "iter/sec",
+            "range": "stddev: 4.687896801992492e-7",
+            "extra": "mean: 2.1988879381303694 usec\nrounds: 100614"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_creation_empty",
+            "value": 138301.9382274831,
+            "unit": "iter/sec",
+            "range": "stddev: 7.431384893607306e-7",
+            "extra": "mean: 7.2305566560836665 usec\nrounds: 27058"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_creation_nested_1",
+            "value": 89161.79388355203,
+            "unit": "iter/sec",
+            "range": "stddev: 7.520901264581019e-7",
+            "extra": "mean: 11.215566179680389 usec\nrounds: 26889"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_creation_nested_2",
+            "value": 73099.12226898775,
+            "unit": "iter/sec",
+            "range": "stddev: 9.246574786017073e-7",
+            "extra": "mean: 13.680054820907873 usec\nrounds: 19664"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_clone",
+            "value": 93489.16855046756,
+            "unit": "iter/sec",
+            "range": "stddev: 0.0000012538679640406563",
+            "extra": "mean: 10.696426286647073 usec\nrounds: 17317"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_getitem[int]",
+            "value": 74529.63019407821,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000002354960862917134",
+            "extra": "mean: 13.417482381114182 usec\nrounds: 17368"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_getitem[slice_int]",
+            "value": 32773.625919222286,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000004853403072500066",
+            "extra": "mean: 30.512339478845487 usec\nrounds: 9547"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_getitem[range]",
+            "value": 17860.01804943473,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000008396422371769982",
+            "extra": "mean: 55.99098484850916 usec\nrounds: 4950"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_getitem[tuple]",
+            "value": 41997.46248806135,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000003878862294560557",
+            "extra": "mean: 23.810962395270206 usec\nrounds: 14546"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_getitem[list]",
+            "value": 19785.464418607058,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000008950449065635528",
+            "extra": "mean: 50.54215452529682 usec\nrounds: 7468"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_setitem_dim[int]",
+            "value": 37310.73616041847,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000003983161665516367",
+            "extra": "mean: 26.801937000129783 usec\nrounds: 1000"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_setitem_dim[slice_int]",
+            "value": 19326.71522180196,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000008083476133098448",
+            "extra": "mean: 51.74185000004172 usec\nrounds: 1000"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_setitem_dim[range]",
+            "value": 13976.014587548274,
+            "unit": "iter/sec",
+            "range": "stddev: 0.0000064770913603231475",
+            "extra": "mean: 71.55115599914552 usec\nrounds: 1000"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_setitem_dim[tuple]",
+            "value": 24947.20423192052,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000006502879829457735",
+            "extra": "mean: 40.084651999620746 usec\nrounds: 1000"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_setitem",
+            "value": 65395.10273244055,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000003455814467959025",
+            "extra": "mean: 15.291664944566714 usec\nrounds: 23614"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_set",
+            "value": 67476.83062412764,
+            "unit": "iter/sec",
+            "range": "stddev: 0.0000044037448679592285",
+            "extra": "mean: 14.819901746280756 usec\nrounds: 23480"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_set_shared",
+            "value": 6203.675477879333,
+            "unit": "iter/sec",
+            "range": "stddev: 0.00006679250556072055",
+            "extra": "mean: 161.1947632602214 usec\nrounds: 3582"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_update",
+            "value": 52811.54076248708,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000004171936296585301",
+            "extra": "mean: 18.935255165104305 usec\nrounds: 25846"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_update_nested",
+            "value": 35760.73204602618,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000005286552418132251",
+            "extra": "mean: 27.963633370618385 usec\nrounds: 18040"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_set_nested",
+            "value": 62375.27799820449,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000002350902684148348",
+            "extra": "mean: 16.031992675508164 usec\nrounds: 21708"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_set_nested_new",
+            "value": 45228.28098603658,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000002631061457287688",
+            "extra": "mean: 22.110059860748013 usec\nrounds: 17524"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_select",
+            "value": 21581.11557341344,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000002774767535226848",
+            "extra": "mean: 46.33680759450342 usec\nrounds: 10613"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_unbind_speed",
+            "value": 3466.6324535160607,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000026014959562822834",
+            "extra": "mean: 288.4643853679214 usec\nrounds: 2761"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_unbind_speed_stack0",
+            "value": 222.2582199934585,
+            "unit": "iter/sec",
+            "range": "stddev: 0.008469616068889742",
+            "extra": "mean: 4.499271163196718 msec\nrounds: 288"
+          },
+          {
+            "name": "benchmarks/common/common_ops_test.py::test_unbind_speed_stack1",
+            "value": 1701213.082786505,
+            "unit": "iter/sec",
+            "range": "stddev: 2.619097126327726e-8",
+            "extra": "mean: 587.81584160053 nsec\nrounds: 80620"
+          },
+          {
+            "name": "benchmarks/common/memmap_benchmarks_test.py::test_creation[device0]",
+            "value": 3419.077447932437,
+            "unit": "iter/sec",
+            "range": "stddev: 0.00013631237526774934",
+            "extra": "mean: 292.4765569743715 usec\nrounds: 1018"
+          },
+          {
+            "name": "benchmarks/common/memmap_benchmarks_test.py::test_creation_from_tensor",
+            "value": 3140.7621172622203,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000020386459879316578",
+            "extra": "mean: 318.3940593602462 usec\nrounds: 2190"
+          },
+          {
+            "name": "benchmarks/common/memmap_benchmarks_test.py::test_add_one[memmap_tensor0]",
+            "value": 39324.39706346922,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000005684960740403288",
+            "extra": "mean: 25.429506227037862 usec\nrounds: 7066"
+          },
+          {
+            "name": "benchmarks/common/memmap_benchmarks_test.py::test_contiguous[memmap_tensor0]",
+            "value": 171545.59875584644,
+            "unit": "iter/sec",
+            "range": "stddev: 5.723502779448721e-7",
+            "extra": "mean: 5.829353870064936 usec\nrounds: 40707"
+          },
+          {
+            "name": "benchmarks/common/memmap_benchmarks_test.py::test_stack[memmap_tensor0]",
+            "value": 53055.052018179886,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000001475201191528637",
+            "extra": "mean: 18.848346424339365 usec\nrounds: 12095"
+          },
+          {
+            "name": "benchmarks/common/memmap_benchmarks_test.py::test_memmaptd_index",
+            "value": 4202.698029591577,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000012507223062138619",
+            "extra": "mean: 237.94238676176815 usec\nrounds: 3082"
+          },
+          {
+            "name": "benchmarks/common/memmap_benchmarks_test.py::test_memmaptd_index_astensor",
+            "value": 1121.4144994869753,
+            "unit": "iter/sec",
+            "range": "stddev: 0.00002603689595410962",
+            "extra": "mean: 891.7309348661713 usec\nrounds: 783"
+          },
+          {
+            "name": "benchmarks/common/memmap_benchmarks_test.py::test_memmaptd_index_op",
+            "value": 456.60207157363135,
+            "unit": "iter/sec",
+            "range": "stddev: 0.00003303783725923713",
+            "extra": "mean: 2.190090808290914 msec\nrounds: 386"
+          },
+          {
+            "name": "benchmarks/common/pytree_benchmarks_test.py::test_reshape_pytree",
+            "value": 43941.91782927643,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000001626313456710349",
+            "extra": "mean: 22.757313503821337 usec\nrounds: 9464"
+          },
+          {
+            "name": "benchmarks/common/pytree_benchmarks_test.py::test_reshape_td",
+            "value": 48554.33059426207,
+            "unit": "iter/sec",
+            "range": "stddev: 0.0000012568461936088252",
+            "extra": "mean: 20.595485258696478 usec\nrounds: 20317"
+          },
+          {
+            "name": "benchmarks/common/pytree_benchmarks_test.py::test_view_pytree",
+            "value": 44052.83796387097,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000002163868248958322",
+            "extra": "mean: 22.700013125604517 usec\nrounds: 16532"
+          },
+          {
+            "name": "benchmarks/common/pytree_benchmarks_test.py::test_view_td",
+            "value": 245603.71949127753,
+            "unit": "iter/sec",
+            "range": "stddev: 2.7829405591052413e-7",
+            "extra": "mean: 4.071599575410805 usec\nrounds: 35320"
+          },
+          {
+            "name": "benchmarks/common/pytree_benchmarks_test.py::test_unbind_pytree",
+            "value": 38485.35989616877,
+            "unit": "iter/sec",
+            "range": "stddev: 0.00000135128972211004",
+            "extra": "mean: 25.983906677706564 usec\nrounds: 10962"
+          },
+          {
+            "name": "benchmarks/common/pytree_benchmarks_test.py::test_unbind_td",
+            "value": 24643.784313000713,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000002045283986479023",
+            "extra": "mean: 40.578183419356364 usec\nrounds: 12207"
+          },
+          {
+            "name": "benchmarks/common/pytree_benchmarks_test.py::test_split_pytree",
+            "value": 38914.00261941368,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000002080534055568855",
+            "extra": "mean: 25.697690617441474 usec\nrounds: 9422"
+          },
+          {
+            "name": "benchmarks/common/pytree_benchmarks_test.py::test_split_td",
+            "value": 13582.104046707278,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000003950017398041949",
+            "extra": "mean: 73.62629505422107 usec\nrounds: 6531"
+          },
+          {
+            "name": "benchmarks/common/pytree_benchmarks_test.py::test_add_pytree",
+            "value": 31210.389375243278,
+            "unit": "iter/sec",
+            "range": "stddev: 0.0000035469633538027035",
+            "extra": "mean: 32.0406127580459 usec\nrounds: 8261"
+          },
+          {
+            "name": "benchmarks/common/pytree_benchmarks_test.py::test_add_td",
+            "value": 22963.575206992693,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000003558356193419997",
+            "extra": "mean: 43.54722603018225 usec\nrounds: 10388"
+          },
+          {
+            "name": "benchmarks/distributed/distributed_benchmark_test.py::test_distributed",
+            "value": 166310.69925198436,
+            "unit": "iter/sec",
+            "range": "stddev: 4.5993024248059834e-7",
+            "extra": "mean: 6.012842255475444 usec\nrounds: 18904"
+          },
+          {
+            "name": "benchmarks/nn/functional_benchmarks_test.py::test_tdmodule",
+            "value": 48981.102450875136,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000009526192861622778",
+            "extra": "mean: 20.416037001268705 usec\nrounds: 1000"
+          },
+          {
+            "name": "benchmarks/nn/functional_benchmarks_test.py::test_tdmodule_dispatch",
+            "value": 26891.009393782402,
+            "unit": "iter/sec",
+            "range": "stddev: 0.00001457523685510716",
+            "extra": "mean: 37.18715000081829 usec\nrounds: 1000"
+          },
+          {
+            "name": "benchmarks/nn/functional_benchmarks_test.py::test_tdseq",
+            "value": 43236.379643585744,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000004704288557695082",
+            "extra": "mean: 23.128670999824408 usec\nrounds: 1000"
+          },
+          {
+            "name": "benchmarks/nn/functional_benchmarks_test.py::test_tdseq_dispatch",
+            "value": 24670.355937594904,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000013119153634220879",
+            "extra": "mean: 40.534477999813134 usec\nrounds: 1000"
+          },
+          {
+            "name": "benchmarks/nn/functional_benchmarks_test.py::test_instantiation_functorch",
+            "value": 772.236908693885,
+            "unit": "iter/sec",
+            "range": "stddev: 0.00005670406284048169",
+            "extra": "mean: 1.2949394010334727 msec\nrounds: 581"
+          },
+          {
+            "name": "benchmarks/nn/functional_benchmarks_test.py::test_instantiation_td",
+            "value": 975.3471988749509,
+            "unit": "iter/sec",
+            "range": "stddev: 0.00008942046031393498",
+            "extra": "mean: 1.0252759234388389 msec\nrounds: 849"
+          },
+          {
+            "name": "benchmarks/nn/functional_benchmarks_test.py::test_exec_functorch",
+            "value": 6690.99607470642,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000007258517749371776",
+            "extra": "mean: 149.4545787853981 usec\nrounds: 2602"
+          },
+          {
+            "name": "benchmarks/nn/functional_benchmarks_test.py::test_exec_td",
+            "value": 6819.051181280562,
+            "unit": "iter/sec",
+            "range": "stddev: 0.0000060327860486782916",
+            "extra": "mean: 146.64796808464607 usec\nrounds: 2256"
+          },
+          {
+            "name": "benchmarks/nn/functional_benchmarks_test.py::test_vmap_mlp_speed[True-True]",
+            "value": 1194.5644127691642,
+            "unit": "iter/sec",
+            "range": "stddev: 0.00003000415225508273",
+            "extra": "mean: 837.1252226423377 usec\nrounds: 1060"
+          },
+          {
+            "name": "benchmarks/nn/functional_benchmarks_test.py::test_vmap_mlp_speed[True-False]",
+            "value": 2207.0938027225125,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000013693024653521914",
+            "extra": "mean: 453.08450359765936 usec\nrounds: 2085"
+          },
+          {
+            "name": "benchmarks/nn/functional_benchmarks_test.py::test_vmap_mlp_speed[False-True]",
+            "value": 1348.1441198119314,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000054653113158087704",
+            "extra": "mean: 741.7604581767577 usec\nrounds: 1327"
+          },
+          {
+            "name": "benchmarks/nn/functional_benchmarks_test.py::test_vmap_mlp_speed[False-False]",
+            "value": 2634.813580756364,
+            "unit": "iter/sec",
+            "range": "stddev: 0.000013393680673658412",
+            "extra": "mean: 379.5334923516428 usec\nrounds: 2484"
+          }
+        ]
       }
     ],
     "GPU Benchmark Results": [