pasteurlabs / tesseract-jaxLinks
Execute + differentiate Tesseracts as part of JAX programs, with full support for function transformations like JIT, grad, and more. β‘
β14Updated this week
Alternatives and similar repositories for tesseract-jax
Users that are interested in tesseract-jax are comparing it to the libraries listed below
Sorting:
- Autodiff-native, self-documenting software components for Simulation Intelligence. π¦β51Updated this week
- Interpolation and function approximation with JAXβ190Updated 3 weeks ago
- Numerical quadrature with JAXβ61Updated this week
- Efficient forward- and reverse-mode sparse Jacobians using Jaxβ57Updated 6 months ago
- This code accompanies the manuscript "A Generative Framework for Probabilistic, Spatiotemporally Coherent Downscaling of Climate Simulatiβ¦β20Updated 4 months ago
- Neural Emulator Architectures in JAX.β16Updated 7 months ago
- Custom Bindings for Enzyme Automatic Differentiation Tool and Interfacing with JAX.β84Updated this week
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Pythonβ478Updated 2 months ago
- Simple Xarray + JAX Integrationβ18Updated 7 months ago
- PyTorch training at CSCSβ15Updated this week
- Pytorch bindings for Fortranβ93Updated last year
- A Python toolkit for coordinating asynchronous and dynamic ensembles of calculations.β70Updated last week
- Automatic-Differentiation-Enabled Plasma Transport in JAXβ31Updated this week
- Jatmos is a scalable simulator for atmospheric dynamics, written in JAX.β18Updated last month
- Visualisation for 4D Volumes (Space + Time) written in WebGPU and Rustβ24Updated 2 weeks ago
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineaxβ445Updated last week
- Incompressible Navier-Stokes solverβ79Updated last week
- Bootcamp notebooksβ54Updated last year
- Implements Optimization and approximate uncertainty quantification algorithms, Ensemble Kalman Inversion, and Ensemble Kalman Processes.β100Updated last week
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) libraryβ95Updated this week
- Orthogonal polynomials with JAXβ22Updated this week
- Quasigeostrophic model in JAX (port of PyQG)β19Updated last week
- Multivariate polynomial interpolation in Python. This is a mirror of the Minterpy repository. Please do not post issues or pull requests β¦β39Updated 8 months ago
- Tools for computing (Lagrangian) coherent structures objectively in Juliaβ24Updated last year
- π Regional-to-global coupled ocean and sea ice simulations based on Oceananigans.β50Updated this week
- Matrix-free linear algebra in JAX.β119Updated last week
- Laplace approximations in JAX.β17Updated this week
- Efficient Differentiable n-d PDE solvers in JAX.β39Updated 7 months ago
- A library for directly calling PyTorch ML models from Fortran.β119Updated this week
- [Neurips 2024] A benchmark suite for autoregressive neural emulation of PDEs. (β₯46 PDEs in 1D, 2D, 3D; Differentiable Physics; Unrolled Tβ¦β73Updated 2 weeks ago