axon
axon copied to clipboard
Nx-powered Neural Networks
I recently ported the YOLOv8 object detection model to Axon, and just wanted to share my experiences with it. https://github.com/hansihe/yolov8_elixir - The deployment story seems a lot better than other...
I think this would enable some pretty interesting functionality, but I haven't yet thought through any potential drawbacks. I will do a detailed proposal soon
There is no need to include these because they are not a part of the optimization process, so it ends up being a waste of memory duplicating them and for...
Related to #582 If your model has multiple inputs but you pass a single tensor it will use that for *every* input in the model. It should raise with an...
I'm still trying to determine if this is Nx or Axon