Phew, it's been a while. I think I only briefly tried it on some basic CNNs, and it worked fine out of the box. But afaik it's mainly for computer vision applications (have been super focused on LLM work recently)

→ View original post on X — @rasbt