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.

Sign up for GitHub

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

Jump to bottom

MPS - Support for Convolutional Neural Network kernels #210

Open
4SAnalyticsnModelling opened this issue Jun 21, 2023 · 1 comment
Open

MPS - Support for Convolutional Neural Network kernels #210

4SAnalyticsnModelling opened this issue Jun 21, 2023 · 1 comment
Labels
help wanted Extra attention is needed libraries Things about libraries and how we use them.

Comments

@4SAnalyticsnModelling
Copy link

Cross posting from Flux.jl FluxML/Flux.jl#2278
@maleadt maleadt changed the title Missing functionalities - Metal.jl with Flux.jl Conv and ConvTranspose layers MPS - Support for Convolutional Neural Network kernels Jun 21, 2023
@maleadt maleadt added enhancement help wanted Extra attention is needed libraries Things about libraries and how we use them. labels Jun 21, 2023
@stemann
Copy link

stemann commented Nov 4, 2024

... someone needs to add support for https://developer.apple.com/documentation/metalperformanceshaders/convolutional_neural_network_kernels and plumb it through to NNlib.jl.

Additional starting points (to start out with something basic - first releases with MPS):

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
help wanted Extra attention is needed libraries Things about libraries and how we use them.
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
4 participants
@maleadt @stemann @tgymnich @4SAnalyticsnModelling