metal

metaljax: JAX'ı Apple Silicon GPU'larında Çalıştıran Metal Backend

metaljax, JAX kodunu Apple Silicon GPU'larında MLX üzerinden çalıştıran açık kaynak bir PJRT eklentisi; JAX test paketinin %98,4'ünü geçiyor.