dblalock / bolt

Related projects ⓘ

Alternatives and complementary repositories for bolt

• facebookincubator / AITemplate
  AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (N…
  ☆4,562Updated last month
• pytorch / torchdynamo
  A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
  ☆1,010Updated 7 months ago
• google / tensorstore
  Library for reading and writing large multi-dimensional arrays.
  ☆1,353Updated this week
• pytorch / glow
  Compiler for Neural Network hardware accelerators
  ☆3,236Updated 6 months ago
• openxla / xla
  A machine learning compiler for GPUs, CPUs, and ML accelerators
  ☆2,712Updated this week
• Tencent / PatrickStar
  PatrickStar enables Larger, Faster, Greener Pretrained Models for NLP and democratizes AI for everyone.
  ☆747Updated last year
• KomputeProject / kompute
  General purpose GPU compute framework built on Vulkan to support 1000s of cross vendor graphics cards (AMD, Qualcomm, NVIDIA & friends). …
  ☆2,007Updated last week
• facebookresearch / fairscale
  PyTorch extensions for high performance and large scale training.
  ☆3,199Updated last week
• iree-org / iree
  A retargetable MLIR-based machine learning compiler and runtime toolkit.
  ☆2,851Updated this week
• google / XNNPACK
  High-efficiency floating-point neural network inference operators for mobile, server, and Web
  ☆1,886Updated this week
• pytorch / FBGEMM
  FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
  ☆1,210Updated this week
• pytorch / data
  A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.
  ☆1,134Updated this week
• rentruewang / koila
  Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code.
  ☆1,820Updated this week
• dmlc / dlpack
  common in-memory tensor structure
  ☆911Updated last month
• triton-lang / triton
  Development repository for the Triton language and compiler
  ☆13,469Updated this week
• pytorch / xla
  Enabling PyTorch on XLA Devices (e.g. Google TPU)
  ☆2,490Updated this week
• tensorflow / mesh
  Mesh TensorFlow: Model Parallelism Made Easier
  ☆1,590Updated last year
• NVIDIA / MatX
  An efficient C++17 GPU numerical computing library with Python-like syntax
  ☆1,220Updated this week
• microsoft / hummingbird
  Hummingbird compiles trained ML models into tensor computation for faster inference.
  ☆3,358Updated last week
• keroro824 / HashingDeepLearning
  Codebase for "SLIDE : In Defense of Smart Algorithms over Hardware Acceleration for Large-Scale Deep Learning Systems"
  ☆1,076Updated 3 years ago
• llvm / torch-mlir
  The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.
  ☆1,356Updated this week
• pytorch / functorch
  functorch is JAX-like composable function transforms for PyTorch.
  ☆1,395Updated this week
• google / highway
  Performance-portable, length-agnostic SIMD with runtime dispatch
  ☆4,224Updated this week
• BaguaSys / bagua
  Bagua Speeds up PyTorch
  ☆876Updated 3 months ago
• facebookresearch / madgrad
  MADGRAD Optimization Method
  ☆803Updated last year
• tensorflow / runtime
  A performant and modular runtime for TensorFlow
  ☆757Updated last month
• NVIDIA / cutlass
  CUDA Templates for Linear Algebra Subroutines
  ☆5,693Updated this week
• ELS-RD / kernl
  Kernl lets you run PyTorch transformer models several times faster on GPU with a single line of code, and is designed to be easily hackab…
  ☆1,535Updated 9 months ago
• tensor-compiler / taco
  The Tensor Algebra Compiler (taco) computes sparse tensor expressions on CPUs and GPUs
  ☆1,256Updated 7 months ago