yiakwy-xpu-ml-framework-team Goto Github PK

alphacodium

amd-ck-fork

Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators

amd-miopen-fork

AMD's Machine Intelligence Library

amd-rocm-docker-fork

Dockerfiles for the various software layers defined in the ROCm software platform

amd-rocm-fork

AMD ROCm™ Software - GitHub Home

dali

A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.

faiss

A library for efficient similarity search and clustering of dense vectors.

fastsam

Fast Segment Anything

flexflow

A distributed deep learning framework.

gbp

Gaussian Belief Propagation for Bundle adjustment and pose graph estimation.

gbp-poplar

Poplar implementation of "Bundle Adjustment on a Graph Processor" (CVPR 2020)

google-tpu-team-mesh-transformer-jax-fork

Model parallel transformers in JAX and Haiku

groqflow

GroqFlow provides an automated tool flow for compiling machine learning and linear algebra workloads into Groq programs and executing those programs on GroqChip™ processors.

hpc-cookbook

Useful tutorials and recipes for developers doing low-level work with the Graphcore IPU

hpc-ipu

Best practice for HPC with IPU backend for scientific/AI(Deep Learning Framework) algorithm and software development

ipudoom

DOOM (1993) on IPU 👿

iputoolkit.jl

IPU programming in Julia

jax-experimental

Experimental JAX for Graphcore IPUs

k8s-nccl-tests

NVIDIA NCCL Tests for Distributed Training

libavif

libavif - Library for encoding and decoding .avif files

lightseq

Official repository for LightSeq: Sequence Level Parallelism for Distributed Training of Long Context Transformers

llama-cpp-python

Python bindings for llama.cpp

llama.cpp

Port of Facebook's LLaMA model in C/C++

llvm-project-fork

Fork of LLVM Project containing a Colossus IPU backend implementation

lm-evaluation-harness

A framework for few-shot evaluation of autoregressive language models.

longctx_chunkllama

Data and code for our paper "Training-Free Long-Context Scaling of Large Language Models"

lwm

mamba

megablocks-fork

megatron-lm

Ongoing research training transformer models at scale