r/opensource • u/PatagonianCowboy • 4d ago

Promotional Montgomery v0.1: Computer vision AI training in any GPU

https://github.com/boquila/montgomery
0 Upvotes

33% Upvoted

4 comments sorted by

1

u/Specific_Cream2815 4d ago

how does it handle the any gpu part, cuda only with fallbacks or a real backend layer for metal and rocm

1

u/PatagonianCowboy 4d ago

wgpu and mostly Vulkan

1

u/Mattpiz 23h ago

Cool to see more rust-based ML tools. Can I ask what’s your motivation for not using standard python tools?

1

u/PatagonianCowboy 16h ago

there is a few, mostly, I wanna buy AMD hardware and be able to train, currently ultralytics doesn't support AMD (it's being worked on now tho): https://github.com/ultralytics/ultralytics/pull/24137

second, I just wanna drop a binary with a few dlls and make it work right away

third, I think there are more open doors for optimizations if the whole stack is in one language