oKatanaaa / CudaCythonSamples
This repository contains examples CUDA usage in Cython code.
☆23Updated 3 years ago
Alternatives and similar repositories for CudaCythonSamples:
Users that are interested in CudaCythonSamples are comparing it to the libraries listed below
- NVIDIA Math Libraries for the Python Ecosystem☆237Updated 2 months ago
- Exploring using stdpar and Cython☆33Updated 4 years ago
- Fusing Taichi into PyTorch☆141Updated 2 years ago
- Example to build PyTorch CUDA extension using CMake (with pybind11 and scikit-build)☆11Updated 4 years ago
- GPU accelerated multigrid library for Python☆55Updated 5 months ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆45Updated this week
- Orthogonal polynomials in all shapes and sizes.☆184Updated last year
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆465Updated this week
- An example combining scikit-build and pybind11☆120Updated this week
- Extending JAX with custom C++ and CUDA code☆386Updated 6 months ago
- Numbast is a tool to build an automated pipeline that converts CUDA APIs into Numba bindings.☆36Updated this week
- Differentiable and accelerated spherical transforms with JAX☆145Updated this week
- A sparse KLU solver for PyTorch.☆62Updated 3 years ago
- A JAX-based research framework for differentiable and parallelizable acoustic simulations, on CPU, GPUs and TPUs☆158Updated 5 months ago
- A nanobind example project☆101Updated last week
- XLB: Accelerated Lattice Boltzmann (XLB) for Physics-based ML☆267Updated this week
- Run PyTorch in JAX. 🤝☆221Updated 2 weeks ago
- 🔗 minimalist ipywidget to interface with any Python vtkRenderWindow☆98Updated 2 years ago
- Example python (numpy) -- CUDA installable package with a C-extension library☆142Updated 5 years ago
- Cubic spline interpolation on multidimensional grids in PyTorch☆26Updated 3 months ago
- Sparse matrix tools extending scipy.sparse, but with incompatible licenses☆164Updated 4 months ago
- Remap, mask, renumber, unique, and in-place transposition of 3D labeled images. Point cloud too.☆53Updated 2 months ago
- numerical optimization in pytorch☆136Updated 9 months ago
- The CUDA target for Numba☆64Updated this week
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆108Updated this week
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆124Updated 5 months ago
- A library of Krylov methods in pure Python☆61Updated 2 years ago
- Template for GPU accelerated python libraries☆46Updated last year
- Nonuniform fast Fourier transforms of types 1 and 2, in 1D, 2D, and 3D, on the GPU☆85Updated last year
- A Python interface to CGAL's meshing tools☆618Updated 7 months ago