Pytorch-world: Building a Modular library for World Models

Hello Everyone,

Since the last few months, I have been studying about world models and along side built a library for learning, training and building new world model algorithms, pytorch-world.

Added a bunch of world model algorithms, components and environments. Still working on adding more. If you find it interesting, I would love to know your thoughts on how I can improve this further or open for collaboration and contributions to make this a better project and useful for everyone researching on world models.

Here’s the link to the repository as well as the Pypi page:
Github repo: https://github.com/ParamThakkar123/pytorch-world
Pypi: https://pypi.org/project/pytorch-world/