Hi, my name is Surya! I really enjoy machine learning and systems. If you find any of my work interesting or just want to say hi, feel free to email me!
- saratoga, ca
-
18:50
(UTC -12:00) - in/suryasub
- @suryaasub
- https://surya-subramanian.github.io/
Pinned Loading
-
pytorch/pytorch
pytorch/pytorch PublicTensors and Dynamic neural networks in Python with strong GPU acceleration
-
meta-pytorch/kraken
meta-pytorch/kraken PublicTriton-based Symmetric Memory operators and examples
-
meta-pytorch/tritonbench
meta-pytorch/tritonbench PublicTritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
-
jax-autodiff
jax-autodiff PublicA JAX-inspired autodiff compiler with optimized function transformations (jit, vmap, grad) and operation fusion.
Python 1
-
diffusion.cu
diffusion.cu PublicImplementation of high-performance diffusion transformer from scratch in CUDA/C++
Jupyter Notebook 8
-
distributed-transformer
distributed-transformer Public"Attention Is All You Need" Transformer paper implemented from scratch in PyTorch with support for distributed training, in both FSDP and DDP.
Python 6
If the problem persists, check the GitHub status page or contact support.