GoMLX est une bibliothèque de machine learning pour Go, accélérée via XLA (le compilateur de TensorFlow/JAX). Elle propose une API de layers (Dense, Conv, Attention, BatchNorm…), un backend interchangeable supportant CPU, GPU NVIDIA et TPU Google, et une intégration Jupyter pour le suivi des métriques en temps réel. Aucune dépendance Python : un simple go get suffit pour démarrer. Des exemples concrets sont disponibles, dont un LLM Gemma 3 et un classifieur CIFAR-10.
Commentaires
Vous devez
vous inscrire
ou
vous connecter
pour poster un commentaire