polvalente

polvalente

Nx Core Team

Nx, EXLA and Torchx 0.12 have just been published!

This update set comes with the new Nx.block abstraction and the EXLA.CustomCall protocol, which together will allow users to provide native C or CUDA implementations for certain sections of Nx code.

EMLX 0.3 (the official MLX backend for Nx) is also out, with:

  1. Support for the more recent concurrency model provided by MLX
  2. EMLX.Fast for a few fused operations (used in the upcoming emlx_axon library, pending Axon and Bumblebee releases in the next few days)
  3. Support for quantized types
  4. A zero-copy implementation of `Nx.to_binary/1`.

Showing Posts 1 to 10

Chapaman

Chapaman

polvalente

polvalente OP

Nx Core Team

emlx_axon has been published!

By using it (see this benchmark for an example), we can accelerate certain Bumblebee models without rewriting code for performance when using EMLX! For example, the Qwen3-0.6B-4bit model goes from ~10 tokens/s to ~42 tokens/s on my machine just by using the provided Axon node rewrites!

manhvu

manhvu

Hi @polvalente, I’m trying to bring Nx + EMLX to iOS. Do you have any suggestions?

Currently, I support inference first, and plan to add basic on-device training later.

polvalente

polvalente OP

Nx Core Team

@manhvu , @GenericJam was able to run EMLX on device, which means all of Nx can run! We’re working to upstream what he needs so that no patches are needed, just config!

manhvu

manhvu

Thanks for the response. I added support for Nx to my framework, and over the next few days I’ll verify and test everything.

For inference, I currently support both the Nx ecosystem and ONNX Runtime (for cross-platform support), but I’ll add Nx support on Android again soon.

GenericJam

GenericJam

The answer is EMLX on iOS and Vulkan on Android. These changes should be upstream shortly.

manhvu

manhvu

EMLX for iOS because it can use Core ML. I’m not sure about Vulkan yet; it doesn’t seem stable enough. I’m still looking for a way to use the mobile GPU on both iOS and Android for on-device training for edge AI.

manhvu

manhvu

Nice, you’ve done that :ok_hand: I looked for the best solution, but it sounds like I need to separate them. Currently, only WebGPU can run on both platforms, so I’ll try to adapt Nx to it.

GenericJam

GenericJam

It looks like @polvalente was right all along and NxIREE is the way to harness Vulkan.

Where Next? Top

Trending in News & Updates Top

pcharbon
:heart::heart::heart::heart::heart::heart::heart::heart::heart::heart::heart::heart::heart::heart::heart::heart::heart::heart::heart::hea...
New
nature
NatureWhistle v0.4.1 is out! :tada: This release is a pretty significant step forward for the project. When I first built NatureWhistle,...
New
webofbits
Aludel 0.7.0 is released :tada: Since 0.5.0, Aludel has grown into a much more complete LLM evaluation toolkit for Elixir and Phoenix app...
New

Other Trending Topics Top

GenericJam
Edit: 2026 May 15 - This post is archived. Mob is alive!! Main docs: mob v0.7.11 — Documentation A bit of explanation for the slightly c...
New
JesseHerrick
Hey, I’m Jesse and I’m the main contributor behind Dexter, a full-featured, lightning-fast Elixir LSP optimized for large codebases. It s...
New
mudasobwa
I am happy to introduce the very α version of the new programming language compiled to BEAM. Welcome Cure. It has literally three kille...
New
marciok
Hi there! We created Gust: A task orchestrator inspired by Airflow. For those who have never heard about Aiflow, it’s a Python-based wor...
New
mhanberg
Hi everyone! The first release candidate for the Expert language server project is now available! We’ve published a press release detai...
New
jimsynz
Beam Bots (or just BB for short) is a framework for building fault-tolerant robotics applications in Elixir using familiar OTP patterns. ...
New

We're in Beta

About us Mission Statement

Options

Thread Display Mode




Thread Preview

Skip Thread Previews