Axiom – C++ tensor library with NumPy API and Metal GPU support
I've been building a C++ tensor library that tries to bring the ergonomics of NumPy/PyTorch to native code. The main goals were: - Familiar API (operator overloading, method chaining, same function names) - Actual GPU acceleration via Metal (not just matmul, but all ops) - Cross-platform with SIMD on x86/ARM/RISC-V via xsimd - einops-style rearrange/reduce patterns It started because I wanted NumPy's API but needed to deploy on edge devices without Python. Ended up going deeper than expected (28k LOC+) into BLAS backends, memory views, and GPU kernels. Some things I'm…
What it does
In the maker’s words, at launch
I've been building a C++ tensor library that tries to bring the ergonomics of NumPy/PyTorch to native code. The main goals were: - Familiar API (operator overloading, method chaining, same function names) - Actual GPU acceleration via Metal (not just matmul, but all ops) - Cross-platform with SIMD on x86/ARM/RISC-V via xsimd - einops-style rearrange/reduce patterns It started because I wanted NumPy's API but needed to deploy on edge devices without Python. Ended up going deeper than expected (28k LOC+) into BLAS backends, memory views, and GPU kernels. Some things I'm genuinely unsure about and would love feedback on: - Does Axiom actually seem useful? - What features might make you use it? GitHub: https://github.com/frikallo/axiom Happy to answer questions about the implementation.
Does the same job
all alternatives →
- POPlaidML, open source deep learning for any GPU2017 · ▲52
Our company Vertex.AI has been working on this for a while but this is the first public release. We're starting with using PlaidML to bring OpenCL support to Keras and more frameworks, platforms, etc are coming. Yes, this means you can use use your AMD GPU for deep learning dev. Sorry, no Mac or Windows support yet although the brave can try building from source (it should work). http://vertex.ai/blog/announcing-plaidml https://github.com/plaidml/plaidml
- TRTensorFlow-Resources – Organized and Useful Resources about TensorFlow2018 · github.com · ▲152
- XCXtensor: C++14 tensor expression library with numpy-style broadcasting2017 · github.com · ▲7
- LSLess Slow C++2025 · github.com · ▲6
Earlier this year, I took a month to reexamine my coding habits and rethink some past design choices. I hope to rewrite and improve my FOSS libraries this year, and I needed answers to a few questions first. Perhaps some of these questions will resonate with others in the community, too. - Are coroutines viable for high-performance work? - Should I use SIMD intrinsics for clarity or drop to assembly for easier library distribution? - Has hardware caught up with vectorized scatter/gather in AVX-512 & SVE? - How do secure enclaves & pointer tagging differ on Intel, Arm, & AMD? - What's…
More dev tools this month
the category →
Open-source GTM skills for technical founders
Dev tools · 29d ago · gtmcofounder.com
OpenTrailPaper is open-source bike computer firmware for the LilyGO T5S3 4.7" E-Paper PRO. It supports offline maps, GPX routes, FIT recording and Bluetooth sensors.
Dev tools · 1d ago · opentrailpaper.com
