YichengDWu / yoho

Related projects ⓘ

Alternatives and complementary repositories for yoho

• iree-org / iree-jax
  ☆48Updated 3 months ago
• leandrolcampos / specials
  Accurate, Hardware Accelerated, Special Functions in Mojo 🔥
  ☆33Updated last month
• nod-ai / PI
  A lightweight MLIR Python frontend with support for PyTorch
  ☆20Updated 2 months ago
• makslevental / mlir-wheels
  ☆21Updated 2 weeks ago
• NVIDIA / numba-cuda
  ☆36Updated this week
• fnands / mimage
  A library for parsing images in Mojo
  ☆13Updated last month
• numba / numba-mlir
  POC work on MLIR backend
  ☆48Updated 2 months ago
• Quansight-Labs / numpy_pytorch_interop
  ☆19Updated last year
• metagraph-dev / mlir-graphblas
  MLIR tools and dialect for GraphBLAS
  ☆16Updated 2 years ago
• Bruce-Lee-LY / cuda_back2back_hgemm
  Use tensor core to calculate back-to-back HGEMM (half-precision general matrix multiplication) with MMA PTX instruction.
  ☆11Updated last year
• EnzymeAD / Enzyme-MLIR-Reverse
  Fork of Enzyme to work on Reverse-Mode Differentiation at the MLIR-level.
  ☆11Updated last year
• EnzymeAD / Enzyme-JAX
  ☆48Updated this week
• ROCm / hip-python
  HIP Python Low-level Bindings
  ☆17Updated this week
• manishucsd / py-codegen
  ☆14Updated last month
• PriNova / MojoPkgWorkflow
  This Repository shows how to use a simple GitHub Action script for compiling a mojo directory into a package.
  ☆12Updated last year
• taichi-dev / taichi_benchmark
  ☆39Updated last year
• williamfgc / simple-gemm
  Collection of simple General Matrix Multiplication - GEMM implementations
  ☆13Updated 8 months ago
• danieljtait / jax_xla_adventures
  Dive into Jax, Flax, XLA and C++
  ☆31Updated 4 years ago
• Python-for-HPC / ramba
  Python combination of Ray and Numba providing compiled distributed arrays, remote functions, and actors.
  ☆32Updated 6 months ago
• makslevental / nelli
  A lightweight, Pythonic, frontend for MLIR
  ☆80Updated last year
• PierrickPochelu / JaxDecompiler
  Jax Decompiler
  ☆13Updated 3 months ago
• Benny-Nottonson / Mojo-Marathons
  ☆29Updated 2 months ago
• microsoft / knossos-ksc
  Compiler with automatic differentiation
  ☆45Updated last year
• pengmai / lagrad
  An MLIR-based source-to-source automatic differentiation system.
  ☆11Updated last year
• JuliaLabs / MLIR.jl
  ☆56Updated 3 weeks ago
• PhilipVinc / Jax.jl
  ☆29Updated 2 years ago
• PhilipVinc / numba4jax
  Use numba-compiled kernels from within Jax
  ☆25Updated last month
• spcl / pymlir
  Python interface for MLIR - the Multi-Level Intermediate Representation
  ☆224Updated 3 weeks ago
• cchan / tccl
  extensible collectives library in triton
  ☆65Updated last month