pasteurlabs / tesseract-jax
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. π¦β50Updated this week
- Numerical quadrature with JAXβ58Updated last week
- Interpolation and function approximation with JAXβ182Updated last week
- Automatic-Differentiation-Enabled Plasma Transport in JAXβ31Updated this week
- UM-Bridge (the UQ and Model Bridge) provides a unified interface for numerical models that is accessible from virtually any programming lβ¦β38Updated last month
- Efficient Differentiable n-d PDE solvers in JAX.β37Updated 6 months ago
- Orthogonal polynomials with JAXβ22Updated last week
- Efficient forward- and reverse-mode sparse Jacobians using Jaxβ57Updated 5 months ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) libraryβ93Updated this week
- Unitful Quantities in JAXβ38Updated this week
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorchβ107Updated 2 months ago
- Multivariate polynomial interpolation in Python. This is a mirror of the Minterpy repository. Please do not post issues or pull requests β¦β39Updated 7 months ago
- Simple Xarray + JAX Integrationβ18Updated 6 months ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizationsβ126Updated 8 months ago
- Stencil computations in JAXβ71Updated last year
- This code accompanies the manuscript "A Generative Framework for Probabilistic, Spatiotemporally Coherent Downscaling of Climate Simulatiβ¦β20Updated 3 months ago
- A small library of paramaterizations and parameter constraints for PyTrees.β29Updated last week
- Custom Bindings for Enzyme Automatic Differentiation Tool and Interfacing with JAX.β78Updated this week
- Material for workshop and autumn school on scientific machine learning 2023β20Updated last year
- Neural Emulator Architectures in JAX.β16Updated 6 months ago
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineaxβ440Updated last week
- Matrix-free linear algebra in JAX.β118Updated 4 months ago
- Visualisation for 4D Volumes (Space + Time) written in WebGPU and Rustβ24Updated last week
- [Neurips 2024] A benchmark suite for autoregressive neural emulation of PDEs. (β₯46 PDEs in 1D, 2D, 3D; Differentiable Physics; Unrolled Tβ¦β70Updated 6 months ago
- β10Updated 2 weeks ago
- differentiable (binned) likelihoods with JAXβ22Updated this week
- Pytorch bindings for Fortranβ93Updated last year
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Pythonβ478Updated last month
- H2 Matrix Packageβ30Updated last year
- PyTorch training at CSCSβ15Updated last year