Machine learning containers

Currently I am using the rocm container provided by hub.docker.com, are there any fedora specific containers for machine learning for rocm or cuda?

Since rocm only provides a rhel install method it would be appreciated a silverblue friendly way of using it.

Hello. Unfortunately, we don’t have such container images. Would you want to help us create them? Could you describe what such container should contain?

I was thinking of a toolbox compatible image with features similar of the rocm image provided by docker (and/or the standard tensorflow image). Currently, there is no clear way of doing machine learning on fedora, especially for amd users.