← Search

Gianni De Fabritiis

5 accepted papers

2024

BricksRL: A Platform for Democratizing Robotics and Reinforcement Learning Research and Education with LEGO

NeurIPS 2024spotlight

We present BricksRL, a platform designed to democratize access to robotics for reinforcement learning research and education. BricksRL facilitates the creation, design, and training of custom LEGO robots in the real world by interfacing them with the TorchRL library for reinforcement learning agents…

2024

TorchRL: A data-driven decision-making library for PyTorch

ICLR 2024spotlight

PyTorch has ascended as a premier machine learning framework, yet it lacks a native and comprehensive library for decision and control tasks suitable for large development teams dealing with complex real-world data and environments. To address this issue, we propose TorchRL, a generalistic control l…

Cited by 46SourcePDFScholar
2023

TensorNet: Cartesian Tensor Representations for Efficient Learning of Molecular Potentials

NeurIPS 2023poster

The development of efficient machine learning models for molecular systems representation is becoming crucial in scientific research. We introduce TensorNet, an innovative O(3)-equivariant message-passing neural network architecture that leverages Cartesian tensor representations. By using Cartesian…

2021

Guided Exploration with Proximal Policy Optimization using a Single Demonstration

ICML 2021spotlight

Solving sparse reward tasks through exploration is one of the major challenges in deep reinforcement learning, especially in three-dimensional, partially-observable environments. Critically, the algorithm proposed in this article is capable of using a single human demonstration to solve hard-explora…