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
- ☆14Updated last year
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆45Updated 2 weeks ago
- Code for the book "The Elements of Differentiable Programming".☆73Updated last week
- Tools for JAX☆45Updated last week
- Official code for UnICORNN (ICML 2021)☆27Updated 3 years ago
- PyTorch implementation for "Probabilistic Circuits for Variational Inference in Discrete Graphical Models", NeurIPS 2020☆16Updated 3 years ago
- An implementation of the 1. Parallel, 2. Streaming, 3. Randomized SVD using MPI4Py☆58Updated 3 years ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆124Updated 5 months ago
- [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…☆17Updated last year
- Computing gradients and Hessians of feed-forward networks with GPU acceleration☆18Updated last year
- Deep learning lectures I am holding for the MSc on Data Science and Scientific Computing☆13Updated 2 years ago
- Turning SymPy expressions into JAX functions☆44Updated 3 years ago
- Curated repository for all the teaching materials (courses, seminars, etc.) provided by the lab and its members.☆8Updated 4 months ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆43Updated last month
- an Open Collaborative project to explore the implications — theoretical or practical — of the PDE perspective of ConvNets☆22Updated last year
- ☆30Updated 5 months ago
- ☆15Updated 4 years ago
- Convolutions and more as einsum for PyTorch☆14Updated 8 months ago
- Riemannian Optimization Using JAX☆48Updated last year
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆93Updated last year
- A lightweight, multithreaded Python package for sketching, column selection, leverage scores and related computations.☆19Updated 5 months ago
- Learning to Learn: Gradient-free Optimization framework☆36Updated 3 years ago
- RKHS feature vectors, operators, and statistical models using JAX for automatic differentiation☆8Updated 3 years ago
- Python Algorithms for Randomized Linear Algebra☆52Updated last year
- simple JAX-/NumPy-based implementations of NGD with exact/approximate Fisher Information Matrix both in parameter-space and function-spac…☆14Updated 4 years ago
- ATHENA: Advanced Techniques for High dimensional parameter spaces to Enhance Numerical Analysis☆51Updated last year
- Tensor decomposition with arbitrary expressions: inner, outer, elementwise operators; nonlinear transformations; and more.☆58Updated 2 years ago
- Bayesian algorithm execution (BAX)☆49Updated 3 years ago
- Repository for the DL Course AA 2020-2021 @ DSSC UniTS☆12Updated 3 years ago
- A library implementing the kernels for and experiments using extrinsic gauge equivariant vector field Gaussian Processes☆25Updated 3 years ago