Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[MLX Backend] CVNets #108

Open
cvnad1 opened this issue Oct 25, 2024 · 1 comment
Open

[MLX Backend] CVNets #108

cvnad1 opened this issue Oct 25, 2024 · 1 comment

Comments

@cvnad1
Copy link

cvnad1 commented Oct 25, 2024

While reviewing the code for the layers used in the CVNets library, my colleague Saanidhya Vats and I observed that certain layers, such as Adaptive Pooling, are inherited from the torch nn module. The maintainer of MLX has suggested that we raise an issue to enable support for CVNets on the MLX backend, which would allow us to take advantage of performance benefits when running on Apple Silicon.

@cvnad1 cvnad1 changed the title [MLX Backend] CV-Nets [MLX Backend] CVNets Oct 25, 2024
@Saanidhyavats
Copy link

Saanidhyavats commented Oct 25, 2024

@sacmehta We opened an issue to propose some new features in MLX. The discussion with the MLX maintainer can be found in this issue: ml-explore/mlx#1512. If CVNets can integrate with MLX-based features, we'd be glad to contribute any missing functionalities to MLX.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants