Sign up to save your podcasts
Or
Share [WIP] CUDA backend by zcbenz · Pull Request #1983 · ml-explore/mlx
Share to email
Share to Facebook
Share to X
Share to email
Share to Facebook
Share to X
Or
[WIP] CUDA backend by zcbenz · Pull Request #1983 · ml-explore/mlx
https://github.com/ml-explore/mlx/pull/1983
This PR is an ongoing effort to add a CUDA backend to MLX, very little things work now but you can run the tutorial example already.
To build and test:
$ cmake . -Bbuild -DMLX_BUILD_CUDA=ON -DMLX_B...
View all episodes
By VoiceFeedhttps://github.com/ml-explore/mlx/pull/1983
This PR is an ongoing effort to add a CUDA backend to MLX, very little things work now but you can run the tutorial example already.
To build and test:
$ cmake . -Bbuild -DMLX_BUILD_CUDA=ON -DMLX_B...