oKatanaaa / CudaCythonSamples

Related projects ⓘ

Alternatives and complementary repositories for CudaCythonSamples

• PWhiddy / pybind11-cuda
  Template for GPU accelerated python libraries
  ☆45Updated last year
• shwina / pyamgx
  GPU accelerated multigrid library for Python
  ☆53Updated last month
• cmpute / pytorch-cmake-example
  Example to build PyTorch CUDA extension using CMake (with pybind11 and scikit-build)
  ☆11Updated 4 years ago
• Autodesk / XLB
  XLB: Accelerated Lattice Boltzmann (XLB) for Physics-based ML
  ☆219Updated this week
• dfm / extending-jax
  Extending JAX with custom C++ and CUDA code
  ☆379Updated 2 months ago
• mpi4jax / mpi4jax
  Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python
  ☆439Updated this week
• wjakob / nanobind_example
  A nanobind example project
  ☆90Updated this week
• NVIDIA / TorchFort
  An Online Deep Learning Interface for HPC programs on NVIDIA GPUs
  ☆155Updated this week
• JAX-DIPS / JAX-DIPS
  JAX-DIPS is a differentiable interfacial PDE solver.
  ☆40Updated last month
• BallisticLA / parla
  Python Algorithms for Randomized Linear Algebra
  ☆46Updated last year
• NVIDIA / nvmath-python
  NVIDIA Math Libraries for the Python Ecosystem
  ☆203Updated 4 months ago
• torstem / demo-cuda-pybind11
  How to use CUDA with Python numpy
  ☆37Updated 6 years ago
• NVIDIA / numbast
  Numbast is a tool to build an automated pipeline that converts CUDA APIs into Numba bindings.
  ☆23Updated this week
• ucl-bug / jaxdf
  A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
  ☆120Updated last month
• Dipolar-Quantum-Gases / jaxfit
  GPU/TPU accelerated nonlinear least-squares curve fitting using JAX
  ☆51Updated last year
• shwina / stdpar-cython
  Exploring using stdpar and Cython
  ☆32Updated 3 years ago
• r-barnes / pytorch_cmake_example
  ☆27Updated 3 years ago
• ucl-bug / jwave
  A JAX-based research framework for differentiable and parallelizable acoustic simulations, on CPU, GPUs and TPUs
  ☆143Updated last month
• astro-informatics / s2fft
  Differentiable and accelerated spherical transforms with JAX
  ☆134Updated 2 weeks ago
• xitorch / xitorch
  Differentiable scientific computing library
  ☆140Updated 2 months ago
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆38Updated last week
• cornellius-gp / linear_operator
  A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch
  ☆93Updated 2 months ago
• esa / torchquad
  Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX
  ☆188Updated 4 months ago
• scikit-sparse / scikit-sparse
  Sparse matrix tools extending scipy.sparse, but with incompatible licenses
  ☆163Updated 3 weeks ago
• rgl-epfl / cholespy
  An easily integrable Cholesky solver on CPU and GPU
  ☆224Updated this week
• BallisticLA / RandLAPACK
  A high-performance C++ library for randomized numerical linear algebra
  ☆60Updated last week
• tumaer / jax-sph
  JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework
  ☆51Updated 2 months ago
• tum-pbs / Solver-in-the-Loop
  Solver-in-the-Loop: Learning from Differentiable Physics to Interact with Iterative PDE-Solvers
  ☆145Updated 2 years ago
• BallisticLA / RandBLAS
  A header-only C++ library for sketching in randomized linear algebra
  ☆79Updated 2 weeks ago
• nicknytko / numml
  ☆14Updated 4 months ago