Pinned Loading

1. TD3-libtorch TD3-libtorch Public

   TD3 reinforcement learning algorithm using libtorch in simple environment

   Makefile 3

2. Biconjugate-gradient--BiCGSTAB-and-Gauss-Jacobi-implementation-on-multi-GPU-system Biconjugate-gradient--BiCGSTAB-and-Gauss-Jacobi-implementation-on-multi-GPU-system Public

   This repository contains implementation for Biconjugate gradient, BiCGSTAB and Jacobi methods for implicit solution of PDE (2D transient heat equation).

   Cuda 2

3. Transformer-Attention-Mechanism-in-CUDA Transformer-Attention-Mechanism-in-CUDA Public

   Custom CUDA kernel for transformer's attention mechanism and integrating it with pytorch

   Cuda 2

4. Custom-CUDA-kernels-with-Neural-Network-Implementation Custom-CUDA-kernels-with-Neural-Network-Implementation Public

   The repository contains custom CUDA kernels for linear layer, softmax and relu which are integrated with python to develop a Neural Network

   Python 1

5. Optimized-matrix-multiplication-in-CUDA Optimized-matrix-multiplication-in-CUDA Public

   The repository contains different ways to implement matrix-matrix multiplication in CUDA starting from basic implementation to using tensor cores in NVIDIA A100 GPUs

   Cuda 1

6. Preconditioned-Conjugate-Gradient-Method-in-CUDA Preconditioned-Conjugate-Gradient-Method-in-CUDA Public

   Preconditioned conjugate gradient method with ILU preconditioner implemented in CUDA

   Cuda 2