What's the best PyTorch model visualization tool?

14 pointsby brainervor 2 Jahren2 comments

I'm looking for a tool like Penzai that was released recently.

What do you guys use?

2 comments

Netron seems to be the best that I've seen so far. https://github.com/lutzroeder/netron

brainervor 2 Jahren

Thanks.

It's great tool.

But I'd like to for xAI.