Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Repository Index
47,133 repositories indexed
1–24 of 26 · 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
ProtTrans is providing state of the art pretrained language models for proteins. ProtTrans was trained on thousands of GPUs from Summit and hundreds of Google TPUs using Transformers Models.
Fast Clojure Matrix Library
C++ library for solving large sparse linear systems with algebraic multigrid method
- amg
- c-plus-plus
- sparse-linear-systems
- opencl
DNABERT: pre-trained Bidirectional Encoder Representations from Transformers model for DNA-language in genome
- kmer
- dnabert-model
- kmer-format
- gpu
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
DiffSharp: Differentiable Functional Programming
- machine-learning
- tensor
- dotnet
- autodiff
Exascale multiphase flow solver — 2025 Gordon Bell Prize Finalist | 200T grid points on 43K+ GPUs
- gpu
- exascale
- cfd
- fortran
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