gahdritz

gahdritz's Stars

• pytorch/pytorch

  Tensors and Dynamic neural networks in Python with strong GPU acceleration

  Language:Python83.6k1.7k46.1k22.5k

• microsoft/DeepSpeed

  DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

  Language:Python35.3k3452.8k4.1k

• ggreer/the_silver_searcher

  A code-searching tool similar to ack, but faster.

  Language:C26.2k4278451.4k

• Dao-AILab/flash-attention

  Fast and memory-efficient exact attention

  Language:Python14k1201.1k1.3k

• google-deepmind/alphafold

  Open source code for AlphaFold.

  Language:Python12.7k2258612.3k

• facebookincubator/AITemplate

  AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (NVIDIA GPU) and MatrixCore (AMD GPU) inference.

  Language:Python4.6k82244369

• srush/Tensor-Puzzles

  Solve puzzles. Improve your pytorch.

  Language:Jupyter Notebook3.3k1320277

• aqlaboratory/openfold

  Trainable, memory-efficient, and GPU-friendly PyTorch reproduction of AlphaFold 2

  Language:Python2.8k47354538

• soedinglab/MMseqs2

  MMseqs2: ultra fast and sensitive search and clustering suite

  Language:C1.4k32829195

• getkeops/keops

  KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows

  Language:Python1.1k1431564

• crowsonkb/v-diffusion-pytorch

  v objective diffusion inference code for PyTorch.

  Language:Python7132917108