pasteurlabs / tesseract-jaxLinks
Execute + differentiate Tesseracts as part of JAX programs, with full support for function transformations like JIT, grad, and more. β‘
β15Updated 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. π¦β57Updated this week
- Numerical quadrature with JAXβ65Updated last week
- Interpolation and function approximation with JAXβ196Updated this week
- Unitful Quantities in JAXβ41Updated 2 weeks ago
- differentiable (binned) likelihoods with JAXβ22Updated this week
- JAX bindings for the NVIDIA cuDecomp libraryβ37Updated this week
- Simple Xarray + JAX Integrationβ19Updated 7 months ago
- Custom Bindings for Enzyme Automatic Differentiation Tool and Interfacing with JAX.β86Updated this week
- Minimal JAX implementation of k-nearest neighbors using a k-d tree.β41Updated last week
- Neural Emulator Architectures in JAX.β16Updated 7 months ago
- A small library of paramaterizations and parameter constraints for PyTrees.β34Updated last month
- Orthogonal polynomials with JAXβ22Updated last week
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) libraryβ97Updated last week
- Finite difference tools in JAXβ20Updated last year
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.β20Updated last year
- Efficient forward- and reverse-mode sparse Jacobians using Jaxβ59Updated last week
- Matrix-free linear algebra in JAX.β125Updated last month
- Multivariate polynomial interpolation in Python. This is a mirror of the Minterpy repository. Please do not post issues or pull requests β¦β40Updated 8 months ago
- Unleash the true power of schedulingβ29Updated 2 months ago
- Efficient Differentiable n-d PDE solvers in JAX.β40Updated 7 months ago
- Manipulating ragged arrays in an Array API compliant way.β44Updated last week
- Multiple dispatch over abstract array types in JAX.β125Updated last week
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilationβ73Updated 2 months ago
- Automatic-Differentiation-Enabled Plasma Transport in JAXβ31Updated this week
- Stencil computations in JAXβ71Updated last year
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Pythonβ481Updated last week
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorchβ111Updated 3 months ago
- Rescure useful code in TensorNetwork by google for further personal usage.β12Updated 6 months ago
- This code accompanies the manuscript "A Generative Framework for Probabilistic, Spatiotemporally Coherent Downscaling of Climate Simulatiβ¦β20Updated 5 months ago
- Implements Optimization and approximate uncertainty quantification algorithms, Ensemble Kalman Inversion, and Ensemble Kalman Processes.β101Updated this week