0.0.3

What's Changed

• [BugFix] tensordict.set(nested_key, value) points to the wrong metadata dict by @vmoens in #40
• [Feature] nested LazyStack indexing by @vmoens in #42
• [Feature] Support nested keys in select method by @tcbegley in #39
• [Feature] Support nested keys in exclude method by @tcbegley in #41
• [Feature] non-strict select by @vmoens in #44
• [BugFix] Use GitHub for flake8 pre-commit hook by @tcbegley in #47
• [BugFix] Accept nested dicts for update by @vmoens in #46
• [BugFix] Fix exclude by @vmoens in #50
• [Feature] Nested keys support for set_default by @tcbegley in #45
• [Feature] TensorDictSequential nested keys by @tcbegley in #49
• [BugFix] Typo in README by @tcbegley in #53
• [Feature] Faster construction by @vmoens in #54
• [BugFix] Nested tensordict collision by @khundman in #51
• [BugFix] Fix memmap creation for nested TDs by @vmoens in #57
• [BugFix] Fix locking mechanism by @vmoens in #58
• [Test] test that cloned locked TDs aren't locked by @vmoens in #59
• [Bugfix] Fix docs build workflow by @tcbegley in #61
• [BugFix] Exclude potentially top-level packages from setup.py by @vmoens in #63
• [Doc] Doc Badge by @vmoens in #64
• [BugFix] Default version to None if not found by @vmoens in #65
• [NOMERGE] Migrate TorchRL to tensordict.nn.TensorDictModule by @tcbegley in #66
• [Formatting] FBCode formatting by @vmoens in #67
• [Feature] TorchRec support: indexing keyedjaggedtensors by @vmoens in #60
• [BugFix] Optional functorch dependency by @vmoens in #68
• [Feature] TensorDict.split method by @wonnor-pro in #56
• [Doc, Test] Some more formatting for FBCode by @vmoens in #71
• [Formatting] Final FBCode formatting by @vmoens in #72
• [Doc] Some more badges by @vmoens in #69
• [Formatting] ignore formatting commits by @vmoens in #73
• [BugFix] Mutable default arguments by @vmoens in #74
• [Formatting] Minor formatting improvements by @vmoens in #75
• [BugFix] functorch fixes for old-deps by @tcbegley in #76
• [Formatting] Fix expand by @vmoens in #78
• Revert "[Formatting] Fix expand" by @vmoens in #79
• [Formatting] Fix expand by @vmoens in #80
• [BugFix] TensorDictSequential inheritance fix by @tcbegley in #81
• [BugFix] TensorDictSequential inheritance fix by @tcbegley in #82
• [Feature] In-place functionalization by @vmoens in #11
• [BugFix] Ensure that all modules are visited during module population by @vmoens in #83
• [Feature] Add leaves_only kwarg to keys / values / items methods by @tcbegley in #84
• [Perf] Faster functional modules by @vmoens in #89
• [Feature] Benchmarks by @vmoens in #90
• [Feature] Lazystack insert / append by @tcbegley in #85
• Revert "[Feature] Benchmarks" by @vmoens in #91
• [Feature] Create a TensorDict from kwargs in TensorDictModule.forward() by @alexanderlobov in #87
• [Docs] Add TensorDict overview to docs by @tcbegley in #93
• [Feature] Benchmarks by @vmoens in #92
• [Versioning] v0.0.1b by @vmoens in #95
• docs: fix esemble -> ensemble typo by @rmax in #96
• [BugFix] Functorch fix by @vmoens in #103
• [Feature] Refactor probabilistic module wrapper by @tcbegley in #104
• [BugFix] patched nn.Module deserialization by @vmoens in #108
• [Feature] Prototype of dataclass like interface for TensorDict by @roccajoseph in #107
• [Feature] Add TensorDict.pop() by @salaxieb in #111
• [Feature] Allow for tensorclass construction with unnamed args by @vmoens in #112
• [Refactor] Refactor dispatch_kwargs for easier usage by @vmoens in #113
• [Feature] Add metaclass for TensorClass instances by @tcbegley in #114
• [Feature] Replace ProbablilisticTensorDictModule with prototype by @tcbegley in #109
• [Refactor] Speedup select method by @vmoens in #120
• [Typo] fist -> first by @tcbegley in #123
• [Feature] Tensorclass device by @tcbegley in #122
• [Feature] Tensorclass updates by @tcbegley in #124
• [BugFix] Fix items_meta, values_meta by @tcbegley in #125
• [Refactor] Various speed improvements by @vmoens in #121
• [BugFix] TorchRec test failure by @vmoens in #126
• [BugFix] Fix functorch vmap imports by @vmoens in #127
• [BugFix] Fix functorch imports by @vmoens in #128
• [Refactor] Remove the unsqueeze for tensors that match the tensordict in shape by @vmoens in #115
• [BugFix] Fix functorch test import by @vmoens in #129
• [Refactor] torch.cat with destination td refactoring by @vmoens in #130
• [BugFix] Fix test_cat device error by @vmoens in #131
• [BugFix] add skipif not _has_functorch by @apbard in #133
• [Formatting] Fix F401 lint advisory by @apbard in #134
• [Feature] Stacking tensors of different shape by @vmoens in #135
• [BugFix] LazyStackedTensorDict indexing along stack_dim by @tcbegley in #138
• [Feature] Support range in indexing operations by @tcbegley in #139
• [BugFix] improving select for LazyStackedTD by @vmoens in #137
• [BugFix] Unbind for lazy stacked TD by @vmoens in #140
• [BugFix] Stack dimension indexing by @tcbegley in #141
• [BugFix] Fix LazyStackedTensorDict.update for LazyStackedTensorDict -> LazyStackedTensorDict by @vmoens in #142
• [BugFix] Prevent calls to get_nestedtensor when stack_dim is not 0 by @tcbegley in #143
• [Doc] Some more doc for tensorclass by @vmoens in #136
• [Versioning] Version 0.0.1c by @vmoens in #144
• [Refactor] Remove prototype import patch by @tcbegley in #117
• [Feature] Add codecov checks by @tcbegley in #86
• [BugFix] Add len method to tensorclass by @tcbegley in #150
• [Doc] Tutorials with sphinx-gallery by @vmoens in #147
• [Doc] More tutorials: ImageNet by @vmoens in #152
• [Doc] Adding external link badges to the documentation by @se-yi in #156
• [Doc] Badges by @vmoens in #158
• [Docs] Typo and formatting in overview example by @tcbegley in #160
• [BugFix] Use distribution kwargs by @vmoens in #161
• [BugFix] Create TensorClass by modifying decorated class in-place by @tcbegley in #165
• [Refactor] Deprecating MetaTensor by @vmoens in #168
• [Feature] TorchSnapshot compatibility by @vmoens in #167
• [Docs] Update to ImageNet tutorial by @tcbegley in #166
• [BugFix] Typo in tensorclass.ne by @tcbegley in #177
• [Refactor] Minor tweak to the data collection in ImageNet tuto by @vmoens in #180
• [Feature] TensorClass __post_init__ support by @tcbegley in #172
• [BugFix, Doc] Minor fix in the doc by @vmoens in #183
• [Versioning] Version 0.0.2a by @vmoens in #184
• [CI, Benchmarks] pytest benchmarks by @mischab in #173
• [CI] pytest benchmarks workflow by @mischab in #174
• [Minor, Doc] Typo by @apbard in #186
• [Refactor] Generate test_tensorclass.py from test_tensorclass_nofuture.py by @tcbegley in #187
• [BugFix] Relax error matching in tensorclass tests by @tcbegley in #188
• [BugFix] Using set_interaction_mode as decorator by @vmoens in #191
• [Feature] apply across multiple TDs by @vmoens in #193
• [BugFix] MemmapTensor from MemmapTensor dir issue by @tcbegley in #192
• [BugFix] Shape change during prealloc by @vmoens in #195
• [BugFix] Incorrect output batch_size with functorch.vmap by @vmoens in #196
• [Feature] More arguments for MemmapTensor construction by @tcbegley in #189
• [Benchmarks] Add MemmapTensor benchmarks by @tcbegley in #197
• [Feature] Added support to save non-tensor data in tensor class by @sreevasthav in #157
• [BugFix] tensorclass __setitem__ must allow non-type stringent writing by @vmoens in #203
• [Test, Bugfix] skip test_outputsize_vmap if no functorch by @apbard in #204
• [CI] Temporarily disable torchrec tests by @apbard in #208
• [Test] MemmapTensor should be cast to tensor and viceversa by @apbard in #206
• [BugFix] Fix _getitem_batch_size in various edge cases. by @tcbegley in #211
• [BugFix] __eq__ refactoring by @vmoens in #210
• [Test] use temp_path instead of tmpdir by @apbard in #212
• [Feature] Implement torch.gather for TensorDict and tensorclass by @vmoens in #213
• [Test, BugFix] add regression test for PR #203 by @apbard in #205
• [Versioning] Version 0.0.2b by @vmoens in #215
• [BugFix] Calling zeros with an empty batch size by @vmoens in #218
• [Feature] Simple symbolic tracer for TensorDictModule and TensorDictSequential by @tcbegley in #162
• [Feature] MemmapTensor.empty_like and TensorDict.memmap_like by @vmoens in #223
• [Feature] set_item works with compatible tensorclass by @apbard in #224
• [CI] increase timeout for tests by @apbard in #225
• Revert "[CI] increase timeout for tests" by @vmoens in #229
• [BugFix] Fix infinite loop in memmap tests by @vmoens in #228
• [Feature] Distributed capabilities by @vmoens in #230
• [BugFix] Fix sorting of keys in send and recv by @vmoens in #234
• [Refactor] Rename set_default -> setdefault by @tcbegley in #235
• [Feature] Full type annotations by @XuehaiPan in #238
• [BugFix] Allow for nested output keys in Probabilitic modules by @vmoens in #239
• [Doc] Port torchrl's tutorials where they belong by @vmoens in #221
• [BugFix] Remove torchrl dependency from tutorials by @vmoens in #245
• [BugFix] Return self from clear_device by @tcbegley in #242
• [Feature] Nesting tensorclass instances in tensordicts by @vmoens in #247
• Revert "[Feature] Nesting tensorclass instances in tensordicts" by @vmoens in #248
• [Feature] Nesting tensorclass instances in tensordicts by @vmoens in #249
• [BugFix] Uncaught wrong instance check in update_at_ by @vmoens in #251
• [BugFix,Feature] Rename with nested keys by @vmoens in #252
• [Docs] Remove unneeded len definition in tutorial by @tcbegley in #254
• [Feature] Make dispatch_kwargs accept different separators by @vmoens in #253
• [BugFix] Memmap-like for lazy stacked tds by @vmoens in #246
• [Refactor] Reduce boilerplate in tensorclass methods by @tcbegley in #250
• [Tests] Add tensorclass benchmarks by @tcbegley in #255
• [Feature] Add dispatch source and dest arguments by @vmoens in #256
• [Doc, BugFix] pin-memory call on cpu only scripts by @vmoens in #257
• [BugFix] Select misses nested keys if missing nested are present by @vmoens in #258
• [Refactor] Refactor set methods by @tcbegley in #259
• [DOC] add clarification comment by @apbard in #263
• [Doc] Remove no_check mentions in docstrings by @vmoens in #262
• [DOC] update docstring for set method by @apbard in #265
• [CI, Test] Retry flaky test by @apbard in #264
• [Refactor] Inplace methods by @tcbegley in #266
• [Quality] More informative error message when an argument is None in TensorDictModule by @vmoens in #261
• [BugFix] Default value for lazy stack by @vmoens in #267
• [Build] missing torchsnapshot dependency by @apbard in #268
• [Test] increase get timeout by @apbard in #269
• [Refactor] Speedup construction by @vmoens in #270
• [Benchmarks] Perform multiple rounds in benchmarks by @tcbegley in #271

New Contributors

• @khundman made their first contribution in #51
• @wonnor-pro made their first contribution in #56
• @alexanderlobov made their first contribution in #87
• @rmax made their first contribution in #96
• @roccajoseph made their first contribution in #107
• @salaxieb made their first contribution in #111
• @apbard made their first contribution in #133
• @se-yi made their first contribution in #156
• @mischab made their first contribution in #173
• @sreevasthav made their first contribution in #157
• @XuehaiPan made their first contribution in #238

Full Changelog: 0.0.1...0.0.3