DSSC-projects / veni
A simple Python package for deep learning using forward automatic differentiation based on JAX.
☆14Updated 2 years ago
Alternatives and similar repositories for veni
Users that are interested in veni are comparing it to the libraries listed below
Sorting:
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆52Updated 2 weeks ago
- [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…☆17Updated last year
- Tools for JAX☆46Updated this week
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆95Updated last year
- Riemannian Optimization Using JAX☆49Updated last year
- Differentiable interface to FEniCS for JAX☆53Updated 3 years ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆126Updated 8 months ago
- Turning SymPy expressions into JAX functions☆45Updated 4 years ago
- Supplementary code for the paper "Meta-Solver for Neural Ordinary Differential Equations" https://arxiv.org/abs/2103.08561☆25Updated 4 years ago
- ☆14Updated 2 years ago
- TorchFSM: Fourier Spectral Method with PyTorch☆43Updated this week
- A library implementing the kernels for and experiments using extrinsic gauge equivariant vector field Gaussian Processes☆25Updated 3 years ago
- Tensor decomposition with arbitrary expressions: inner, outer, elementwise operators; nonlinear transformations; and more.☆58Updated 2 years ago
- Computing gradients and Hessians of feed-forward networks with GPU acceleration☆18Updated last year
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆45Updated 4 months ago
- Inference on non-linear dynamical systems written in JAX☆11Updated 4 years ago
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆119Updated last month
- Convolutions and more as einsum for PyTorch☆16Updated 11 months ago
- Auto-differentiable and hardware-accelerated force density method☆88Updated 3 months ago
- Code for Lie Symmetries SSL paper☆22Updated last year
- Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…☆11Updated 2 years ago
- Stencil computations in JAX☆71Updated last year
- A software package for flexible HPC GPs☆16Updated this week
- ☆107Updated 4 years ago
- Code for the book "The Elements of Differentiable Programming".☆84Updated last month
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆46Updated last month
- Bayesian algorithm execution (BAX)☆49Updated 3 years ago
- Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"☆26Updated last year
- Modular pipeline for design assistance. PYPi Package: https://pypi.org/project/vai-lab☆14Updated last year
- Python Algorithms for Randomized Linear Algebra☆54Updated 2 years ago