[the__alchemist]

I'm writing scientific software that has components (molecular dynamics) that are much faster on GPU. I'm using CUDA only, as it's the eaisiest to code for. I'd assumed this meant no-go on ARM Macs. Does this news make that false?

[wmf]

This driver doesn't support CUDA.

[ksec]

This comment should be pinned at the top.

[brcmthrowaway]

Isnt mlx a cuda translation later?

[ykl]

No, MLX is nothing like a Cuda translation layer at all. It’d be more accurate to describe MLX as a NumPy translation layer; it lets you write high level code dealing with NumPy style arrays and under the hood will use a Metal GPU or CUDA GPU for execution. It doesn’t translate existing CUDA code to run on non-CUDA devices.

[superb_dev]

My understanding is that MLX is Apple’s CUDA, so a CUDA translation layer would target MLX

[ykl]

No, it’s not. MLX is Apple’s NumPy more or less.

[wmf]

Does tinygrad support MLX?