Jie Xu's repositories
cccl
CUDA C++ Core Libraries
codelldb
A native debugger extension for VSCode based on LLDB
CoDiPack
Fast gradient evaluation in C++ based on Expression Templates.
diffusers
🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch
Fastor
A lightweight high performance tensor algebra framework for modern C++
gfem2d-matlab
A finite-element-like method for 2D plasmonics, especially graphene, implemented in Matlab
ginkgo
Numerical linear algebra software package
HIP
HIP: C++ Heterogeneous-Compute Interface for Portability
K-Litho
K-Litho is a toolset designed for lithography simulation. It enables optical image calculations using TCC and SOCS, as well as SOCS kernel derivation.
libxsmm
Library for specialized dense and sparse matrix operations, and deep learning primitives.
llvm
Intel staging area for llvm.org contribution. Home for Intel LLVM-based projects.
MeDiPack
Message Differentiation Package
OpDiLib
Universal add-on for operator overloading AD tools that enables the differentiation of OpenMP parallel codes.
precice
A coupling library for partitioned multi-physics simulations, including, but not restricted to fluid-structure interaction and conjugate heat transfer simulations.
pytorch-image-models
PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNet-V3/V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more
pytorch-lightning
Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.
stable-diffusion
A latent text-to-image diffusion model
superlu_dist
Distributed memory, MPI based SuperLU
ThunderKittens
Tile primitives for speedy kernels
vllm
A high-throughput and memory-efficient inference and serving engine for LLMs
wildmeshing-toolkit
Toolkit for Meshing in the Wild