mlx
mlx copied to clipboard
MLX: An array framework for Apple silicon
Is there any way to import pretrained SOTA CV models (e.g. MobileNet) instead of creating toy CV models? I didn't find examples. Thanks for any response
Thanks to the mlx team for creating and sharing mlx. I have managed to get a small CIFAR-10 image classification CNN up and running rather quickly in mlx (inspired by...
First of all - this library is a dream come true for a grad student. I've been using [Taichi](https://github.com/taichi-dev/taichi) for doing things on the GPU up to this point (which...
Not sure if this is an issue of mine, but I'm unable to use mx.matmul or use the @ operator on two mx.arrays, it throws the following error: ` libc++abi:...
I kept encountering the below error while trying the stable diffusion sample in [mlx-examples](https://github.com/ml-explore/mlx-examples) on an 8GB M2 Mac Mini here. After some investigation (detailed here: https://github.com/ml-explore/mlx-examples/issues/21) I found changing...
[The Python array API standard](https://data-apis.org/array-api/2022.12/) standardises common functionality across Python array/tensor libraries. NumPy, PyTorch and CuPy are planning to have full implementations, and Dask and JAX also have implementations in...
When running mlx-examples/mnist on MacOS 14.1.1 VM running via Parallels 19.1.1: ``` user@Users-Virtual-Machine mnist % python main.py --gpu *** Terminating app due to uncaught exception 'NSInvalidArgumentException', reason: '-[AppleParavirtDevice newArgumentEncoderWithLayout:]: unrecognized...
we can start there: https://uvadlc-notebooks.readthedocs.io/en/latest/tutorial_notebooks/JAX/tutorial7/GNN_overview.html or here: https://danielegrattarola.github.io/posts/2021-03-12/gnn-lecture-part-2.html but ideally GIN, DMPNN and AttentiveFP are part of the best models so far: https://github.com/aimat-lab/gcnn_keras/tree/master/kgcnn/literature
Thanks for the nice framework! However, the medical imaging community is missing 3d operations, such as `conv3d`.