tanmoyio

tanmoyio

accelerate

🚀 A simple way to train and use PyTorch models with multi-GPU, TPU, mixed-precision

alphafold

Open source code for AlphaFold.

babymap

Manifold projection for Babies

babynet

efficient training inference of DL models

backend

[Graphsitry + Triton] backend for accelerated graph inference ⚡️

codeforces-haskell

Codeforces solutions in Haskell

dfencoder

dgl

Python package built to ease deep learning on graph, on top of existing DL frameworks.

dm-haiku

JAX-based neural network library

dotfiles

Environment dotfiles

flash-attention

github-nvim-theme

Github theme for Neovim and Terminals written in Lua

hivemind

Decentralized deep learning in PyTorch. Built to train models on thousands of volunteers across the world.

ipython

Official repository for IPython itself. Other repos in the IPython organization contain things like the website, documentation builds, etc.

lightning

The lightweight PyTorch wrapper for high-performance AI research. Scale your models, not the boilerplate.

MarkovJunior

Probabilistic PL based on pattern matching and constraint propagation, 148 examples

Megatron-DeepSpeed

Ongoing research training transformer language models at scale, including: BERT & GPT-2

Morpheus

Morpheus SDK

MRC

Morpheus Runtime Core (MRC)

numpy

The fundamental package for scientific computing with Python.

open-gpu-kernel-modules

NVIDIA Linux open GPU kernel module source

openpilot

openpilot is an open source driver assistance system. openpilot performs the functions of Automated Lane Centering and Adaptive Cruise Control for over 150 supported car makes and models.

pygraphistry

PyGraphistry is a Python library to quickly load, shape, embed, and explore big graphs with the GPU-accelerated Graphistry visual graph analyzer

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

pytorch_geometric

Graph Neural Network Library for PyTorch

tinycat

tinygrad

You like pytorch? You like micrograd? You love tinygrad! ❤️

tinyvoice

Letting computers listen to you and really care

TransformerEngine

A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper GPUs, to provide better performance with lower memory utilization in both training and inference.