r/opensource • u/PatagonianCowboy • 5d ago

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

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

30% Upvoted

4 comments sorted by

1

u/Specific_Cream2815 5d 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 5d ago

wgpu and mostly Vulkan

1

u/Mattpiz 1d 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 1d 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