EnzymeAD / Enzyme-JAXLinks
Custom Bindings for Enzyme Automatic Differentiation Tool and Interfacing with JAX.
☆108Updated this week
Alternatives and similar repositories for Enzyme-JAX
Users that are interested in Enzyme-JAX are comparing it to the libraries listed below
Sorting:
- Use numba-compiled kernels from within Jax☆31Updated 7 months ago
- Multiple dispatch over abstract array types in JAX.☆136Updated last week
- Stencil computations in JAX☆71Updated last month
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆108Updated last week
- Sparse and Structured Tensor Compiler☆231Updated last month
- Numerical quadrature with JAX☆73Updated last week
- Flexible and performant GEMM kernels in Julia☆83Updated this week
- A small library of paramaterizations and parameter constraints for PyTrees.☆47Updated 7 months ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆75Updated 2 months ago
- Linnea is an experimental tool for the automatic generation of optimized code for linear algebra problems.☆72Updated 4 months ago
- Programming Gemm Kernels on NVIDIA GPUs with Tensor Cores in Julia☆43Updated last week
- ☆28Updated 3 years ago
- Julia library to manipulate MLIR dialects.☆65Updated 10 months ago
- A zoo of implementations of differential equation problems in NumPy and JAX. Oscillators, chemical reactions, n-body problems, epidemiolo…☆17Updated 2 years ago
- CPU/GPU portable `parallel_for` `parallel_reduce` in Julia for productive science. Funded by the US DOE Advanced Scientific Computing Res…☆56Updated last week
- Matrix-free linear algebra in JAX.☆153Updated 3 weeks ago
- Orthogonal polynomials with JAX☆26Updated last week
- Turning SymPy expressions into JAX functions☆45Updated 4 years ago
- Optimize Julia Functions With MLIR and XLA for High-Performance Execution on CPU, GPU, TPU and more.☆275Updated this week
- Structure Preserving Machine Learning Models in Julia☆54Updated last week
- Differentiable interface to Firedrake for JAX☆15Updated 4 years ago
- "Maybe we have our own magic."☆48Updated 5 years ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆131Updated last year
- Bayesian inference on spatial and spatiotemporal data, faster than you can say "Cholesky!"☆12Updated 2 weeks ago
- Ridiculously fast symbolic expressions☆130Updated 2 weeks ago
- Taylor-mode automatic differentiation for higher-order derivatives☆81Updated 3 weeks ago
- No need to train, he's a smooth operator☆45Updated last week
- Implementations of single and multi-ellipsoid nested sampling☆47Updated last month
- A Julia package for hierarchically semiseparable (HSS) matrices.☆26Updated 2 years ago
- A dead simple Python package for creating custom JAX pytree objects☆66Updated last year