0.2.0

New features

• [Feature] First class dim compatibility by @vmoens in #525
• [Feature] CompositeDistribution by @vmoens in #517
• [Feature] torch.where by @vmoens in #507
• [Feature] dense_stack_tds by @matteobettini in #506
• [Feature] TensorDictParams by @vmoens in #500
• [Feature] Hooks and Buffers for TensorDictParams by @vmoens in #502
• [Feature] PyTree compatibility by @vmoens in #501
• [Feature] from_module by @vmoens in #423
• [Feature] Prevent lock/unlock conflicts for stacked and nested tensordicts by @vmoens in #436
• [Feature] Lock check decorator by @vmoens in #440
• [Feature] Cache for locked tensordicts by @vmoens in #438
• [Feature] tensordict.transpose by @vmoens in #467

What's Changed

• [Benchmark] Benchmark flatten / unflatten by @vmoens in #439

• [Benchmark] Benchmark items, values and keys by @vmoens in #428

• [Benchmark] Benchmark leaf membership by @vmoens in #462

• [Benchmark] Benchmark set nested by @vmoens in #459

• [Benchmark] Benchmark to(device) by @vmoens in #385

• [Benchmark] Benchmark unbind by @vmoens in #470

• [Benchmark] Benchmark vmap and stacked tds by @vmoens in #430

• [Benchmark] Fix GPU benchmark by @vmoens in #386

• [Benchmark] Memmap tensordict benchmarks by @vmoens in #432

• [Benchmark] More benchmarks for get by @vmoens in #477

• [Benchmark] More benchmarks for locked tds by @vmoens in #437

• [Benchmark] Total time per benchmark for efficiency by @vmoens in #446

• [BugFix] Fix EnsembleModule parameters exposure by @vmoens in #508

• [BugFix] Fix None device when loading by @vmoens in #425

• [BugFix] Fix benchmarks in PR comparison by @vmoens in #442

• [BugFix] Fix benchmarks in PR comparison by @vmoens in #447

• [BugFix] Fix error for incongruent devices in load_state_dict by @vmoens in #529

• [BugFix] Fix error in state_dict tests by @vmoens in #531

• [BugFix] Fix het lazy stack ops by @vmoens in #416

• [BugFix] Fix hook call by @vmoens in #457

• [BugFix] Fix indexing of lazy stacks by @vmoens in #488

• [BugFix] Fix lazy stack / stack_onto + masking lazy stacks by @vmoens in #497

• [BugFix] Fix lazy stack lock by @vmoens in #452

• [BugFix] Fix lazy-stack to call to to_tensordict() by @vmoens in #519

• [BugFix] Fix logical operations in MemmapTensor by @vmoens in #484

• [BugFix] Fix max length by @vmoens in #408

• [BugFix] Fix opt deps by @vmoens in #464

• [BugFix] Fix params.clone by @vmoens in #509

• [BugFix] Fix pickle PR by @vmoens in #397

• [BugFix] Fix regex in tensorclass error checks by @vmoens in #398

• [BugFix] Fix state-dict by @vmoens in #528

• [BugFix] Fix test rename_key by @vmoens in #441

• [BugFix] Fix unbind by @vmoens in #471

• [BugFix] Fix wheels and nightly by @vmoens in #417

• [BugFix] Fix writing on stacked tds in vmap by @vmoens in #456

• [BugFix] Heterogeneous lazy stack apply(_) by @vmoens in #400

• [BugFix] Keyword arguments with dispatch + select keys by @vmoens in #409

• [BugFix] Nested keys to probabilistic modules by @matteobettini in #479

• [BugFix] Overload expand by @vmoens in #530

• [BugFix] Repr of het stacked tds by @vmoens in #399

• [BugFix] Retrieve all keys for LazyStackedTensorDict by @matteobettini in #512

• [BugFix] Torch 1.13 compat by @vmoens in #421

• [BugFix] Torch 1.13 compat by @vmoens in #424

• [BugFix] Torch 1.13 compat by @vmoens in #429

• [BugFix] Unflatten names by @vmoens in #472

• [BugFix] fix infinite loop by @xmaples in #392

• [BugFix] from_module without detach by @vmoens in #466

• [BugFix] stateless lstms by @vmoens in #475

• [Bugfix] Indexing memmaps with torch.Tensor by @tcbegley in #383

• [Bugfix] fix getitem by @apbard in #384

• [CI] Concurrency on gha by @vmoens in #387

• [CI] Fix Doc CI bis by @matteobettini in #490

• [CI] Fix doc CI by @matteobettini in #489

• [CI] Python 3.11 compatibility by @vmoens in #516

• [CI] Removing myself from notified users in benchmarks by @tcbegley in #483

• [CI] nightly run by @vmoens in #422

• [CI] nightly test by @vmoens in #420

• [Doc] Fix missing packaging package by @vmoens in #449

• [Doc] Update citation by @vmoens in #404

• [Feature] Add EnsembleModule by @smorad in #485

• [Feature] Add reset_parameters support to TensorDictModuleBase by @smorad in #476

• [Feature] Better error message in tdmodule by @vmoens in #390

• [Feature] Cat LazyStackedTensorDicts by @matteobettini in #499

• [Feature] Create memmaps folders with makesdir by @vmoens in #431

• [Feature] Functional reset_parameters by @smorad in #478

• [Feature] Kwargs for tensordict modules by @vmoens in #405

• [Feature] Module serialization by @vmoens in #396

• [Feature] Module to add state independent normal scale by @albertbou92 in #515

• [Feature] Print lazy fields by @matteobettini in #493

• [Feature] Registering a tensordict as a module buffer by @vmoens in #395

• [Feature] Support classmethods in tensorclass by @vmoens in #448

• [Feature] TensorDict.map by @vmoens in #518

• [Feature] Torch 1.13 compatibility by @vmoens in #414

• [Feature] Unravel nested keys by @vmoens in #403

• [Feature] as_decorator and enter exit methods by @vmoens in #455

• [Feature] batch_dims for from_dict by @vmoens in #460

• [Feature] unravel_key_list by @vmoens in #474

• [Minor] Add dist/ to gitignore. by @skandermoalla in #492

• [Minor] Better errors for kwargs and key selection by @vmoens in #415

• [Minor] Fix typo in _propagate_unlock by @vmoens in #443

• [Minor] Key filtering error fixes during deletion by @vmoens in #481

• [Misc] Move to PyTorch compat by @vmoens in #533

• [Perf] Perf improvements by @vmoens in #412

• [Performance] Better names handling in LazyStackTD by @vmoens in #482

• [Performance] Faster functional by @vmoens in #418

• [Performance] Faster keys by @vmoens in #427

• [Performance] Faster vmap by @vmoens in #426

• [Performance] Refactor get by @vmoens in #445

• [Performance] Speed improvements by @vmoens in #410

• [Performance] use dict for modules in swap states by @vmoens in #419

• [Refactor] Better key membership check by @vmoens in #461

• [Refactor] Better key membership check by @vmoens in #465

• [Refactor] Better keys for cache to reduce mem footprint by @vmoens in #454

• [Refactor] Bottom-up get_tuple by @vmoens in #451

• [Refactor] Faster benchmarks by @vmoens in #453

• [Refactor] Faster creation on device by @vmoens in #389

• [Refactor] Faster indexing of memmap tensors by @vmoens in #433

• [Refactor] Fix warnings by @vmoens in #402

• [Refactor] Minor refactorings to tensorclass by @vmoens in #435

• [Refactor] Refactor to create empty tds where necessary by @vmoens in #522

• [Refactor] Refactor to() by @vmoens in #381

• [Refactor] Simplify exclude by @vmoens in #391

• [Refactor] TensorDict.to() signature by @vmoens in #520

• [Refactor] _set_str and _set_tuple by @vmoens in #480

• [Refactor] unravel_key fixes by @vmoens in #473

• [Setup] Update setup.py python versions by @vmoens in #521

• [Test,CI,Feature] Total time per test by @vmoens in #407

• New Contributors

• @xmaples made their first contribution in #392

• @smorad made their first contribution in #476

• @matteobettini made their first contribution in #479

• @skandermoalla made their first contribution in #492

• @albertbou92 made their first contribution in #515

Full Changelog: v0.1.2...v0.2.0