Search Results
5 results for “library”
DeepSpeed
DeepSpeed is an open-source deep learning optimisation library developed by Microsoft that enables efficient distributed training and inference of large-scale neural networks through memory and compute optimisations.
FAISS
FAISS (Facebook AI Similarity Search) is an open-source library developed by Meta AI for efficient similarity search and clustering of dense vectors at scale, enabling fast nearest-neighbour retrieval across millions or billions of embeddings.
JAX
JAX is an open-source numerical computing library from Google that combines NumPy-style array programming with automatic differentiation and just-in-time compilation, used to train large-scale machine learning models on GPUs and TPUs.
Scikit-learn
Scikit-learn is an open-source Python library for classical machine learning, providing accessible and consistent implementations of classification, regression, clustering, and data-preprocessing algorithms built on NumPy and SciPy.
vLLM
vLLM is an open-source library for fast and memory-efficient large language model inference and serving, built around the PagedAttention algorithm for optimised GPU memory management.