amx

Apple AMX Instruction Set (by corsix)

Amx Alternatives

Similar projects and alternatives to amx

NOTE: The number of mentions on this list indicates mentions on common posts plus user suggested alternatives. Hence, a higher number means a better amx alternative or higher similarity.

amx reviews and mentions

Posts with mentions or reviews of amx. We have used some of these posts to build our list of alternatives and similar projects. The last one was on 2024-02-28.
  • Optimize sgemm on RISC-V platform
    6 projects | news.ycombinator.com | 28 Feb 2024
    I am talking about the matrix/vector coprocessor (AMX). You can find some reverse-engineered documentation here: https://github.com/corsix/amx

    On M3 a singe matrix block can achieve ~ 1TFLOP on DGEMM, I assume it will be closer to 4TFLOPS for SGEMM. The Max variants have two such blocks. Didn't do precise benchmarking myself, but switching Python/R matrix libraries to use Apple's BLAS result in 5-6x perf improvement on matrix heavy code for me.

  • Intel AMX
    4 projects | news.ycombinator.com | 19 Jan 2024
    It's really cool. I hope it becomes more common for training/inference/numerics capable accelerators to be included in consumer hardware.

    Apple's AMX is really under-documented, while the instructions were reverse engineered, Virtually no benchmarks are available comparing current chip generations, models and variants.

    https://github.com/corsix/amx

  • Why do x86 processors take up so much energy when compared to ARM?
    1 project | /r/hardware | 8 Dec 2023
  • Bfloat16 support coming to Apple's Metal and PyTorch [video]
    1 project | news.ycombinator.com | 3 Jul 2023
    Visible in the unofficial documentation for AMX instructions too - M2 only bf16 functionality - https://github.com/corsix/amx/blob/main/matfp.md
  • LLaMA-7B in Pure C++ with full Apple Silicon support
    19 projects | news.ycombinator.com | 10 Mar 2023
    Confusingly there are 2 mechanisms to do matrix operations on the new apple hardware - AMX (https://github.com/corsix/amx) - and the ANE (apple neural engine) - which is enabled by CoreML. This code does not run on the neural engine but the author has a branch for his whisper.cpp project which uses it here: https://github.com/ggerganov/whisper.cpp/pull/566 - so it may not be long before we see it applied here as well. All of this is to say that it actually could get significantly faster if some of this work was able to be handed to the ANE with CoreML.
  • Linux 6.2: The first mainstream Linux kernel for Apple M1 chips arrives
    7 projects | news.ycombinator.com | 20 Feb 2023
    really? seems pretty well documented here: https://github.com/corsix/amx
  • AMX: The Secret Apple M1 Coprocessor
    1 project | /r/apple | 14 Dec 2022
    Article is almost two years old, and has a huge correction at the bottom. It's just a proprietary ISA extension, there's even a repo documenting what's been reverse engineered.
  • corsix/amx: Apple AMX Instruction Set
    1 project | /r/programming | 9 Dec 2022
  • Show HN: Port of OpenAI's Whisper model in C/C++
    9 projects | news.ycombinator.com | 6 Dec 2022
    You are correct, in that those are the four

    My understanding is that the AMX is more tightly wound with the CPU, ultimately being accessible via an instruction set (https://github.com/corsix/amx), and it is useful if you need to do matrix multiplications interleaved with other CPU tasks. A common example would be a VIO loop or something where you want that data in the CPU caches.

    The GPU and Neural Engine are not that – they take some time to set up and initialize. They also can parallelize tasks to a much higher degree. The GPU is more generalizable, because you can write compute shaders to do anything in parallel, but it uses a lot of resources. I'll have to check out the PR to see how exactly the MPS shaders match up with the task at hand, because you could also consider writing Metal compute shaders by hand.

    I know the least about the ANE, but it has specific hardware for running ML models, and you have to process the weights ahead of time to make sure they are in the right format. It can run ML models very efficiently and is the most battery friendly.

  • Ask HN: Are there any undocumented ISA extensions used in Linux systems?
    1 project | news.ycombinator.com | 19 Oct 2022
    If someone were to build a Linux system with proprietary ISA extensions, how would they do it given Linux is open source? Are there any examples of this being done? Would it be possible at all?

    I got inspiration from this (https://github.com/corsix/amx) and I wondered if someone has done it before on a Linux-based system. I understand a userspace library could be created to access those instructions from userspace, but how would then they be implemented in the kernel? Through a proprietary kernel module built using a custom compiler? Or is that not needed at all and the library could just run on the processor taking advantage of the proprietary extensions?

  • A note from our sponsor - WorkOS
    workos.com | 25 Apr 2024
    The APIs are flexible and easy-to-use, supporting authentication, user identity, and complex enterprise features like SSO and SCIM provisioning. Learn more →

Stats

Basic amx repo stats
18
843
4.1
about 2 months ago

corsix/amx is an open source project licensed under MIT License which is an OSI approved license.

The primary programming language of amx is C.


Sponsored
SaaSHub - Software Alternatives and Reviews
SaaSHub helps you find the best software and product alternatives
www.saashub.com