A multilingual chatbot for climate risk awareness in Toronto

Aya Expedition model

AMD ROCm™ Software - GitHub Home

A scalable generative AI framework built for researchers and developers working on Large Language Models, Multimodal, and Speech AI (Automatic Speech Recognition and Text-to-Speech)

Ocean is the in-house framework for Computer Vision (CV) and Augmented Reality (AR) applications at Meta. It is platform independent and is mainly implemented in C/C++.

To speed up Long-context LLMs' inference, approximate and dynamic sparse calculate the attention, which reduces inference latency by up to 10x for pre-filling on an A100 while maintaining accuracy.

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

🤘 TT-NN operator library, and TT-Metalium low level kernel programming model.

An open-source nuclear reactor analysis automation framework that helps design teams increase efficiency and quality

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

The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

OpenVDB - Sparse volume data structure and tools

The LLVM Project is a collection of modular and reusable compiler and toolchain technologies.

A collection of high-quality models for the MuJoCo physics engine, curated by Google DeepMind.

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

PennyLane is a cross-platform Python library for quantum computing, quantum machine learning, and quantum chemistry. Train a quantum computer the same way as a neural network.

StarCraft II Learning Environment

Multi-Joint dynamics with Contact. A general purpose physics simulator.

Optax is a gradient processing and optimization library for JAX.

Code to accompany "A Method for Animating Children's Drawings of the Human Figure"

TorchMultimodal is a PyTorch library for training state-of-the-art multimodal multi-task models at scale.

Tracking Any Point (TAP)

Hybrid ML + physics model of the Earth's atmosphere

SONAR, a new multilingual and multimodal fixed-size sentence embedding space, with a full suite of speech and text encoders and decoders.

NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source components of TensorRT.