Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Repository Index
47,133 repositories indexed
1–24 of 24 · Most stars
Play with fluids in your browser (works even on mobile)
- fluid
- webgl
- simulation
- gpu
Burn is a next generation tensor library and Deep Learning Framework that doesn't compromise on flexibility, efficiency and portability.
- autodiff
- deep-learning
- machine-learning
- rust
Suite of tools for deploying and training deep learning models using the JVM. Highlights include model import for keras, tensorflow, and onnx/pytorch, a modular and tiny c++ library for running math code and a java based math library on top of the core c++ library. Also includes samediff: a pytorch/tensorflow like library for running deep learn...
- java
- gpu
- deeplearning
- neural-nets
Open3D: A Modern Library for 3D Data Processing
- mesh-processing
- computer-graphics
- opengl
- cpp
NumPy & SciPy for GPU
- cuda
- cudnn
- cublas
- cusolver
CUDA Templates and Python DSLs for High-Performance Linear Algebra
- cuda
- deep-learning
- deep-learning-library
- cpp
ArrayFire: a general purpose GPU library.
Trainable, memory-efficient, and GPU-friendly PyTorch reproduction of AlphaFold 2
- pytorch
- protein-structure
- alphafold2
Multi-dimensional arrays (tensors) and numerical definitions for Elixir
A collection of GICP-based fast point cloud registration algorithms
- point-cloud
- pcl
- gicp
- icp
An efficient C++20 GPU numerical computing library with Python-like syntax
C++ library for solving large sparse linear systems with algebraic multigrid method
- amg
- c-plus-plus
- sparse-linear-systems
- opencl
⚡ Datoviz: high-performance GPU rendering for scientific data visualization
- visualization
- graphics
- data-visualization
- scientific-visualization
VexCL is a C++ vector expression template library for OpenCL/CUDA/OpenMP
- opencl
- cuda
- c-plus-plus
- gpgpu
Numerical linear algebra software package
- linear-algebra
- hpc
- gpu-computing
- cuda
Exascale multiphase flow solver — 2025 Gordon Bell Prize Finalist | 200T grid points on 43K+ GPUs
- gpu
- exascale
- cfd
- fortran
High-performance Bayesian Data Analysis on the GPU in Clojure
- clojure
- bayesian-inference
- bayesian-data-analysis
- gpu-computing
CEED Library: Code for Efficient Extensible Discretizations
- ceed
- high-order
- exascale-computing
- api
Open source math and physics engine that executes MATLAB-syntax code. Features a modern compiler, automatic GPU acceleration for large calculations, and support for math on geometry / FEA calculations.
- matlab
- rust
- jit-compiler
- linear-algebra
Scientific computing in pure Crystal
- crystal
- linear-algebra
- machine-learning
- autograd
👩💻 Matrix and other math, accelerated by WebGPU and WASM
Go's answer to pandas + scikit-learn: a batteries-included data analysis library with dataframes, statistics, machine learning, quant, and GPU acceleration. Run Python with seamless two-way variable sharing, or explore data straight from the Insyra CLI.
OpenCL based GPU accelerated SPH fluid simulation library