Viewing a single comment thread. View all comments

patrickkidger t1_irake85 wrote

Reply to comment by Small_Stand_8716 in [R] Introduction to Diffusion Models in JAX by jakiwjakiw

Have you seen eqxvision? It's still relatively nascent but already has a fair amount.

3

Small_Stand_8716 t1_ire7h35 wrote

Thank you, eqxvision fits the bill for me actually. For the most part, I need pre-trained backbones (usually just RegNet, EfficientNetV2, or ViT) and code the rest myself (YOLO, DETR, and so forth). Thanks again!

1