#machine-learning repositories

MNN: A blazing-fast, lightweight inference engine battle-tested by Alibaba, powering high-performance on-device LLMs and Edge AI.

Open-source deep-learning framework for building, training, and fine-tuning deep learning models using state-of-the-art Physics-ML methods

torchax is a PyTorch frontend for JAX. It gives JAX the ability to author JAX programs using familiar PyTorch syntax. It also provides JAX-Pytorch ...

High-quality implementations of standard and SOTA methods on a variety of tasks.

Trax — Deep Learning with Clear Code and Speed

BioNeMo Framework: For building and adapting AI models in drug discovery at scale

A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit and 4-bit floating point (FP8 and FP4) precision on Hopper, Ada...

CausalMatch is a Bytedance research project aimed at integrating cutting-edge machine learning and econometrics methods to bring about automation i...

A retargetable MLIR-based machine learning compiler and runtime toolkit.

Tensors and Dynamic neural networks in Python with strong GPU acceleration

🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for b...

Cross-platform, customizable ML solutions for live and streaming media.

Google Research

An Open Source Machine Learning Framework for Everyone

Core ML tools contain supporting tools for Core ML model conversion, editing, and validation.

Running BERT without Padding

Train machine learning models within a 🐳 Docker container using 🧠 Amazon SageMaker.

CLIP (Contrastive Language-Image Pretraining), Predict the most relevant text snippet given an image

A library for training and deploying machine learning models on Amazon SageMaker

The official Python client for the Hugging Face Hub.

Datasets, Transforms and Models specific to Computer Vision

Manipulating Python Programs

Example 📓 Jupyter notebooks that demonstrate how to build, train, and deploy machine learning models using 🧠 Amazon SageMaker.

🤗 Evaluate: A library for easily evaluating machine learning models and datasets.