Avatar for Shizheng Wen from gravatar.com

Shizheng Wen

Username    wshizhen
Date joined   Joined

3 projects

torch-sla

Last released

PyTorch Sparse Linear Algebra - Differentiable sparse solvers with CUDA support

flashslice

Last released

Fused Triton kernels for the slice/deslice bottleneck of physics-attention, and the Transolver layer they accelerate.

tensormesh-fem

Last released

Differentiable Finite Element Method Library for PyTorch
Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page