oKatanaaa / CudaCythonSamples
This repository contains examples CUDA usage in Cython code.
☆20Updated 3 years ago
Related projects ⓘ
Alternatives and complementary repositories for CudaCythonSamples
- Template for GPU accelerated python libraries☆45Updated last year
- Extending JAX with custom C++ and CUDA code☆378Updated 3 months ago
- GPU accelerated multigrid library for Python☆53Updated last month
- Fusing Taichi into PyTorch☆135Updated last year
- An Online Deep Learning Interface for HPC programs on NVIDIA GPUs☆156Updated 2 weeks ago
- Example to build PyTorch CUDA extension using CMake (with pybind11 and scikit-build)☆11Updated 4 years ago
- A nanobind example project☆91Updated last week
- Numbast is a tool to build an automated pipeline that converts CUDA APIs into Numba bindings.☆27Updated this week
- An open source library for the GPU-implementation of L-BFGS-B algorithm☆116Updated last year
- XLB: Accelerated Lattice Boltzmann (XLB) for Physics-based ML☆232Updated 2 weeks ago
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆445Updated 2 weeks ago
- JAX-DIPS is a differentiable interfacial PDE solver.☆40Updated 2 months ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆39Updated last week
- NVIDIA Math Libraries for the Python Ecosystem☆205Updated this week
- How to use CUDA with Python numpy☆37Updated 6 years ago
- numerical optimization in pytorch☆134Updated 6 months ago
- ☆27Updated 3 years ago
- Simple GPU rendering of scientific data with Pytorch, Jax, CuPy, and Warp backends.☆21Updated last year
- GPU-Accelerated multigrid solver for Poisson's equation in 2D☆20Updated 3 years ago
- A sparse KLU solver for PyTorch.☆62Updated 2 years ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆121Updated 2 months ago
- Compute interesting points, areas, and volumes in simplex meshes of any dimension.☆104Updated 5 months ago
- An easily integrable Cholesky solver on CPU and GPU☆225Updated 2 weeks ago
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆190Updated 4 months ago
- 3D Gaussian Splatting in JAX☆54Updated 5 months ago
- Exploring using stdpar and Cython☆32Updated 4 years ago
- LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite☆58Updated 4 months ago
- Orthogonal polynomials in all shapes and sizes.☆181Updated 9 months ago
- Solver-in-the-Loop: Learning from Differentiable Physics to Interact with Iterative PDE-Solvers☆146Updated 2 years ago
- JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework☆53Updated 2 months ago