core/go-mlx
8
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity Actions
go-mlx/internal/metal
History
Snider ceb966b66b feat(metal): expose model metadata via Info() 
Return architecture, vocab size, layer count, hidden dimension, and
quantisation config (bits + group size) for loaded models.

Gemma3-1B 4-bit: arch=gemma3, vocab=262144, layers=26, hidden=1152,
quant=4-bit/group64.

Co-Authored-By: Virgil <virgil@lethean.io>
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-02-19 23:36:23 +00:00
..
array.go fix(metal): address 4 minor code review items 2026-02-19 21:36:40 +00:00
array_test.go fix(metal): address 4 minor code review items 2026-02-19 21:36:40 +00:00
backend.go fix(metal): address 3 critical code review items 2026-02-19 21:24:10 +00:00
batch.go feat(metal): add inference metrics (timing, throughput, memory) 2026-02-19 23:34:40 +00:00
batch_test.go feat(metal): implement batch inference (Classify, BatchGenerate) 2026-02-19 23:28:15 +00:00
bench_test.go bench(metal): add 29 benchmarks baselined on M3 Ultra 2026-02-19 20:47:25 +00:00
cache.go refactor(metal): flatten model, tokenizer, sample, cache into internal/metal 2026-02-19 19:51:14 +00:00
cache_test.go refactor(metal): move all tests to internal/metal (148 tests passing) 2026-02-19 20:00:02 +00:00
close.go feat(metal): deterministic Close() and Array.Iter() 2026-02-19 21:09:39 +00:00
close_test.go feat(metal): deterministic Close() and Array.Iter() 2026-02-19 21:09:39 +00:00
compile.go fix(metal): address 5 important code review items 2026-02-19 21:31:45 +00:00
dtype.go refactor(metal): move dtype, array, metal, stream to internal/metal 2026-02-19 19:34:38 +00:00
error_test.go feat(metal): implement batch inference (Classify, BatchGenerate) 2026-02-19 23:28:15 +00:00
fast.go fix(metal): address 4 minor code review items 2026-02-19 21:36:40 +00:00
fast_test.go refactor(metal): move all tests to internal/metal (148 tests passing) 2026-02-19 20:00:02 +00:00
gemma3.go feat(metal): implement batch inference (Classify, BatchGenerate) 2026-02-19 23:28:15 +00:00
generate.go feat(metal): expose model metadata via Info() 2026-02-19 23:36:23 +00:00
grad.go fix(metal): address 4 minor code review items 2026-02-19 21:36:40 +00:00
grad_test.go test(metal): validate gradient checkpointing with real model (Phase 3) 2026-02-19 23:11:15 +00:00
io.go fix(metal): error handling audit — propagate MLX errors instead of swallowing 2026-02-19 20:59:37 +00:00
iter_test.go feat(metal): deterministic Close() and Array.Iter() 2026-02-19 21:09:39 +00:00
lora.go feat(metal): add mixed precision training via LoRAConfig.DType (Phase 3) 2026-02-19 23:13:49 +00:00
lora_test.go fix(metal): error handling audit — propagate MLX errors instead of swallowing 2026-02-19 20:59:37 +00:00
metal.go fix(metal): address 4 minor code review items 2026-02-19 21:36:40 +00:00
model.go feat(metal): implement batch inference (Classify, BatchGenerate) 2026-02-19 23:28:15 +00:00
model_test.go test(metal): add model loading robustness tests (Phase 2) 2026-02-19 21:49:07 +00:00
nn.go refactor(metal): move nn, io, grad, lora, optim to internal/metal 2026-02-19 19:39:58 +00:00
nn_test.go refactor(metal): move all tests to internal/metal (148 tests passing) 2026-02-19 20:00:02 +00:00
ops.go fix(metal): address 4 minor code review items 2026-02-19 21:36:40 +00:00
ops_test.go feat(metal): bind CumSum, implement TopP and MinP sampling 2026-02-19 20:39:44 +00:00
optim.go refactor(metal): move nn, io, grad, lora, optim to internal/metal 2026-02-19 19:39:58 +00:00
optim_test.go refactor(metal): move all tests to internal/metal (148 tests passing) 2026-02-19 20:00:02 +00:00
qwen3.go feat(metal): implement batch inference (Classify, BatchGenerate) 2026-02-19 23:28:15 +00:00
random.go fix(metal): address 4 minor code review items 2026-02-19 21:36:40 +00:00
sample.go feat(metal): bind CumSum, implement TopP and MinP sampling 2026-02-19 20:39:44 +00:00
sample_test.go fix(metal): address 5 important code review items 2026-02-19 21:31:45 +00:00
slice.go fix(metal): address 4 minor code review items 2026-02-19 21:36:40 +00:00
stream.go fix(metal): address 5 important code review items 2026-02-19 21:31:45 +00:00
tokenizer.go feat(metal): add Llama 3 model support (Llama 3.1 8B validated) 2026-02-19 23:06:43 +00:00
tokenizer_test.go fix(metal): address 5 important code review items 2026-02-19 21:31:45 +00:00
train_test.go feat(metal): add mixed precision training via LoRAConfig.DType (Phase 3) 2026-02-19 23:13:49 +00:00