Highlights
- Pro
Starred repositories
A JAX backend for Apple Metal Performance Shaders (MPS), enabling GPU-accelerated JAX computations on Apple Silicon.
Shadow any website for offline viewing, with the JavaScript stripped out
A basic Python implementation of the deflated continuation algorithm.
Scripts to download and assemble printable versions of Roche's famous biochemical pathways poster
A Python framework for GPU-accelerated simulation, robotics, and machine learning.
Collection of high-performance, CUDA-accelerated fluid dynamics and physics simulators, including SPH, hypersonic flow, and reaction-diffusion systems.
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python ⚡
Slides for a summer school course at the EMS School on Mathematical Modelling, Numerical Analysis and Scientific Computing
This repository contains examples of the proximal Galerkin finite element method and other proximal numerical methods derived from the LVPP algorithm
Differentiable Finite Element Method with JAX
Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.
An open-source Mathematica Kernel. This repository contains the Python modules for WL Built-in functions, variables, core primitives, e.g. Symbol, a parser to create Expressions, and an evaluator t…
Efficient Differentiable n-d PDE Solvers in JAX.
Scalene: a high-performance, high-precision CPU, GPU, and memory profiler for Python with AI-powered optimization proposals
Computations and statistics on manifolds with geometric structures.
The official SuiteSparse library: a suite of sparse matrix algorithms authored or co-authored by Tim Davis, Texas A&M University.
Synthesizing Graphics Programs for Scientific Figures and Sketches with TikZ.
JVector: the most advanced embedded vector search engine
Solve puzzles. Improve your pytorch.
interactive visualization of 5 popular gradient descent methods with step-by-step illustration and hyperparameter tuning UI
A JAX research toolkit for building, editing, and visualizing neural networks.