Pinned Loading

1. pytorch/pytorch pytorch/pytorch Public

   Tensors and Dynamic neural networks in Python with strong GPU acceleration

   Python 91.6k 24.7k

2. facebookincubator/AITemplate facebookincubator/AITemplate Public

   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.

   Python 4.7k 383

3. Weiszfeld's Algorithm to numerically... Weiszfeld's Algorithm to numerically approximate the geometric median for a list of ML model weights.

   1

   def average(models, eps=1e-5, tol=1e-6):

   2

       """

   3

       Computes geometric median of the model weights using Weiszfeld's Algorithm

   4

       """

   5

       models = [model.get_weights() for model in models]

4. Very simple Python interpreter writt... Very simple Python interpreter written in Python

   1

   # ref: https://www.aosabook.org/en/500L/a-python-interpreter-written-in-python.html

   2

   from typing import NamedTuple, Optional, Union, List

   3

   from collections import deque

   4

   from dataclasses import dataclass

   5