Daniel Hiltgen 539741199e mlx: perf improvements (#14768) ...

* mlx: perf improvements

Fix nn.go to call mlx_fast_layer_norm instead of manually implementing (mean,
subtract, variance, rsqrt, multiply, add — 6 ops)

Fix llama.go, gemma3.go to remove RepeatKV to tile K/V tensors to match the Q
head count, since scaled_dot_product_attention natively handles GQA (it just
requires n_q_heads % n_kv_heads == 0)

* review comments

2026-03-12 12:01:28 -07:00

..

generator

ci: Fix windows build (#14754)

2026-03-09 19:27:59 -07:00

include/mlx/c

MLX: add header vendoring and remove go build tag (#14642)

2026-03-09 17:24:45 -07:00

.gitignore

Add MLX runner with GLM4-MoE-Lite model support (#14185)

2026-02-10 14:57:57 -08:00

act.go

MLX: add header vendoring and remove go build tag (#14642)

2026-03-09 17:24:45 -07:00

array_test.go

MLX: add header vendoring and remove go build tag (#14642)

2026-03-09 17:24:45 -07:00

array.go

MLX: add header vendoring and remove go build tag (#14642)

2026-03-09 17:24:45 -07:00

CMakeLists.txt

MLX: add header vendoring and remove go build tag (#14642)

2026-03-09 17:24:45 -07:00

dtype.go

MLX: add header vendoring and remove go build tag (#14642)

2026-03-09 17:24:45 -07:00

dynamic.c

mlx: remove noisy error output from dynamic library loading (#14346)

2026-02-20 23:46:07 -08:00

dynamic.go

mlx: perf improvements (#14768)

2026-03-12 12:01:28 -07:00

dynamic.h

MLX: add header vendoring and remove go build tag (#14642)

2026-03-09 17:24:45 -07:00

fast.go

MLX: add header vendoring and remove go build tag (#14642)

2026-03-09 17:24:45 -07:00

gated_delta.go

MLX: add header vendoring and remove go build tag (#14642)

2026-03-09 17:24:45 -07:00

generated.c

MLX: add header vendoring and remove go build tag (#14642)

2026-03-09 17:24:45 -07:00

generated.h

update mlx-c bindings to 0.5.0 (#14380)

2026-02-23 16:44:29 -08:00

io.go

MLX: add header vendoring and remove go build tag (#14642)

2026-03-09 17:24:45 -07:00

memory.go

MLX: add header vendoring and remove go build tag (#14642)

2026-03-09 17:24:45 -07:00

mlx.go

ci: Fix windows build (#14754)

2026-03-09 19:27:59 -07:00

nn.go

MLX: add header vendoring and remove go build tag (#14642)

2026-03-09 17:24:45 -07:00

ops_extra.go

mlx: perf improvements (#14768)

2026-03-12 12:01:28 -07:00

ops.go

MLX: add header vendoring and remove go build tag (#14642)

2026-03-09 17:24:45 -07:00

random.go

MLX: add header vendoring and remove go build tag (#14642)

2026-03-09 17:24:45 -07:00

slice.go

MLX: add header vendoring and remove go build tag (#14642)

2026-03-09 17:24:45 -07:00

stream.go

MLX: add header vendoring and remove go build tag (#14642)

2026-03-09 17:24:45 -07:00