DSSC-projects / veniLinks
A simple Python package for deep learning using forward automatic differentiation based on JAX.
☆15Updated 2 years ago
Alternatives and similar repositories for veni
Users that are interested in veni are comparing it to the libraries listed below
Sorting:
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆136Updated 2 months ago
- Riemannian Optimization Using JAX☆51Updated last year
- Gradient-based constrained optimization for JAX☆35Updated 2 years ago
- This repository provides open-source code for sparse continuous distributions and corresponding Fenchel-Young losses.☆16Updated 2 years ago
- Computing gradients and Hessians of feed-forward networks with GPU acceleration☆20Updated last year
- ☆105Updated last year
- Tools for JAX☆49Updated last week
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- Supplementary code for the paper "Meta-Solver for Neural Ordinary Differential Equations" https://arxiv.org/abs/2103.08561☆25Updated 4 years ago
- Pytorch implementation of SuperPolyak subgradient method.☆43Updated 2 years ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆56Updated 2 weeks ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- Turning SymPy expressions into JAX functions☆45Updated 4 years ago
- Inference on non-linear dynamical systems written in JAX☆11Updated 4 years ago
- ☆80Updated 3 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆40Updated 4 years ago
- Bayesian algorithm execution (BAX)☆50Updated 3 years ago
- Parameter-Free Optimizers for Pytorch☆130Updated last year
- Pytrees + dataclasses ❤️☆67Updated 3 months ago
- Simple and extensible hypergradient for PyTorch☆18Updated 2 years ago
- Composable kernels for scikit-learn implemented in JAX.☆44Updated 4 years ago
- ☆60Updated 3 years ago
- Repository of machine learning benchmarks☆39Updated this week
- [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…☆17Updated 2 years ago
- Official code for UnICORNN (ICML 2021)☆27Updated 3 years ago
- Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)☆55Updated 3 weeks ago
- This repository contains code released by DiffEqML Research☆90Updated 3 years ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆42Updated 5 years ago
- ☆20Updated 3 years ago
- Stochastic Automatic Differentiation library for PyTorch.☆205Updated 11 months ago