Jie Xu's repositories
accelerate
🚀 A simple way to train and use PyTorch models with multi-GPU, TPU, mixed-precision
annotated_deep_learning_paper_implementations
🧑🏫 60 Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit, ...), optimizers (adam, adabelief, sophia, ...), gans(cyclegan, stylegan2, ...), 🎮 reinforcement learning (ppo, dqn), capsnet, distillation, ... 🧠
arrayfire-binary-python-wrapper
A python thin wrapper for ArrayFire library
ComfyUI
The most powerful and modular stable diffusion GUI, api and backend with a graph/nodes interface.
CUDALibrarySamples
CUDA Library Samples
cuml
cuML - RAPIDS Machine Learning Library
cuspatial
CUDA-accelerated GIS and spatiotemporal algorithms
dedalus
A flexible framework for solving PDEs with modern spectral methods.
devsim
TCAD Semiconductor Device Simulator
fealpy
Finite Element Analysis Library in Python
FMM3D
Flatiron Institute Fast Multipole Libraries --- This codebase is a set of libraries to compute N-body interactions governed by the Laplace and Helmholtz equations, to a specified precision, in three dimensions, on a multi-core shared-memory machine.
ginkgo
Numerical linear algebra software package
gloo
Collective communications library with various primitives for multi-machine training.
GSLIB
Mirror of gslib - Gather/Scatter library for nearest neighbor data exchange. Please use the official repository, https://github.com/gslib/gslib, to create issues and pull requests.
MoM_Basics.jl
A Julia package for meshes and basis functions of Method of Moments (MoM) and Multi-level Fast Multipole Algorithm (MLFMA).
MoM_Kernels.jl
A Julia package for kernel functions of Method of Moments (MoM) and Multi-level Fast Multipole Algorithm (MLFMA).
oneAPI-samples
Samples for Intel® oneAPI Toolkits
oneMKL
oneAPI Math Kernel Library (oneMKL) Interfaces
PyTMM
Transfer Matrix Method implemetation & RefractiveIndex.info database browser
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-optimizer
torch-optimizer -- collection of optimizers for Pytorch
torch-harmonics
Differentiable spherical harmonic transforms and spherical convolutions in PyTorch
TorchLitho
The lithography simulator
transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
trl
Train transformer language models with reinforcement learning.