Ben Albrecht's starred repositories
transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
latent-diffusion
High-Resolution Image Synthesis with Latent Diffusion Models
DALLE2-pytorch
Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis neural network, in Pytorch
imagen-pytorch
Implementation of Imagen, Google's Text-to-Image Neural Network, in Pytorch
AITemplate
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.
awesome-jax
JAX - A curated list of resources https://github.com/google/jax
ml_collections
ML Collections is a library of Python Collections designed for ML use cases.
ml-compiler-opt
Infrastructure for Machine Learning Guided Optimization (MLGO) in LLVM.
import-tracker
Python utility for tracking third party dependencies within a library
actions-comment-pull-request
GitHub action to comment pull request
pull-request-comment-trigger
A github action for detecting a "trigger" in a pull request description or comment
ml-deployment-k8s-fastapi
This project shows how to serve an ONNX-optimized image classification model as a web service with FastAPI, Docker, and Kubernetes.
smartsim-openFOAM
Using OpenFOAM with SmartSim. This example implements a streaming version of the TensorFlowFOAM work