NVIDIA / mcmc-bnn-exampleLinks
Reference CUDA implementation of training a small Bayesian neural network (BNN) using MCMC
☆18Updated 4 years ago
Alternatives and similar repositories for mcmc-bnn-example
Users that are interested in mcmc-bnn-example are comparing it to the libraries listed below
Sorting:
- The Foundation for All Legate Libraries☆235Updated this week
- Python SYCL bindings and SYCL-based Python Array API library☆121Updated this week
- Kernel Tuner☆383Updated this week
- NumPy and SciPy on Multi-Node Multi-GPU systems☆966Updated this week
- Data Parallel Extension for NumPy☆121Updated this week
- Extending JAX with custom C++ and CUDA code☆403Updated last year
- Numbast is a tool to build an automated pipeline that converts CUDA APIs into Numba bindings.☆56Updated last week
- The NVIDIA® Tools Extension SDK (NVTX) is a C-based Application Programming Interface (API) for annotating events, code ranges, and resou…☆507Updated last week
- Data Parallel Extension for Numba☆90Updated 4 months ago
- Distributed multigrid linear solver library on GPU☆646Updated last month
- The CUDA target for Numba☆251Updated this week
- Numerical linear algebra software package☆555Updated last week
- RAPIDS Memory Manager☆681Updated this week
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆513Updated this week
- Some CUDA design patterns and a bit of template magic for CUDA☆158Updated 2 years ago
- An example combining scikit-build and pybind11☆141Updated 2 weeks ago
- QUDA is a library for performing calculations in lattice QCD on GPUs.☆342Updated this week
- JAX-Toolbox☆382Updated this week
- ☆103Updated last week
- Fast and full-featured Matrix Market I/O library for C++, Python, and R☆86Updated last year
- Exploring using stdpar and Cython☆34Updated 5 years ago
- Tutorial for wrapping C++ library into Python using pybind11 and CMake☆152Updated 2 years ago
- ☆618Updated this week
- An efficient C++20 GPU numerical computing library with Python-like syntax☆1,402Updated last week
- A suite of benchmarks for CPU and GPU performance of the most popular high-performance libraries for Python☆334Updated last year
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆380Updated this week
- A header-only C++ library for sketching in randomized linear algebra☆96Updated this week
- Archived implementation of BLAS using the SYCL open standard. See oneMath for a replacement.☆260Updated last year
- ☆55Updated 2 months ago
- Nvidia contributed CUDA tutorial for Numba☆265Updated 3 years ago