All of a sudden I care about Compute. When will AMD become competitive in this not-so-niche market?

• AITemplate

  37 4,448 9.1 Python

  AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (NVIDIA GPU) and MatrixCore (AMD GPU) inference.

  

For cross-vendor AI, Meta's AITemplate may be the best, as it has support for both CUDA and HIP. However, I haven't tested it so can't tell how easy it is to get running.

• Pytorch

  336 77,783 10.0 Python

  Tensors and Dynamic neural networks in Python with strong GPU acceleration

  

This might not apply in your circumstance where you are required to use dockerized model, but if one just wants pytorch working in a python3 virtual environment, I wonder if instruction on the front page of pytorch.org "Just Works"?

• WorkOS

  workos.com sponsored

  The modern identity platform for B2B SaaS. The APIs are flexible and easy-to-use, supporting authentication, user identity, and complex enterprise features like SSO and SCIM provisioning.

  

• stable-diffusion

  382 65,243 0.0 Jupyter Notebook

  A latent text-to-image diffusion model

  

Setting up Stable Diffusion directory and conda environment: Go to the main stable diffusion repo https://github.com/CompVis/stable-diffusion download their environment.yaml file (raw), put it into the StableDiffusion folder (for the above command it'll be at $HOME/StableDiffusion)

Suggest a related project