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