Skip to content

tomazzzzzx/gpu-inference-engine

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 
 
 

Repository files navigation

GPU Inference Engine

High-throughput LLM inference optimized for AMD Instinct MI300X.

Features

  • Batched inference with dynamic padding
  • KV-cache for continuous batching
  • ROCm 6.x native support
  • Flash attention v2 for CDNA3
  • Tensor parallelism (multi-GPU)
  • OpenAI-compatible API server

Benchmarks

Model MI300X H100 A100
Llama-3-70B 142 tok/s 138 tok/s 89 tok/s
Mixtral-8x22B 98 tok/s 91 tok/s 62 tok/s

License: Apache 2.0

About

High-throughput LLM inference engine optimized for AMD Instinct MI300X GPUs

Topics

Resources

Stars

0 stars

Watchers

0 watching

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages