clayton-r / symtorch

Related projects ⓘ

Alternatives and complementary repositories for symtorch

• patrick-kidger / sympytorch
  Turning SymPy expressions into PyTorch modules.
  ☆142Updated last year
• willieneis / bayesian-algorithm-execution
  Bayesian algorithm execution (BAX)
  ☆48Updated 3 years ago
• Information-Fusion-Lab-Umass / NuX
  Normalizing Flows using JAX
  ☆82Updated 11 months ago
• facebookresearch / rcpm
  Riemannian Convex Potential Maps
  ☆68Updated last year
• MilesCranmer / sympy2jax
  Turning SymPy expressions into JAX functions
  ☆42Updated 3 years ago
• Zymrael / torchSODE
  PyTorch block-diagonal ODE CUDA solver, designed for gradient-based optimization
  ☆16Updated 4 years ago
• SaitejaUtpala / rieoptax
  Riemannian Optimization Using JAX
  ☆45Updated last year
• imbirik / LieStationaryKernels
  Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"
  ☆41Updated last year
• cornellius-gp / linear_operator
  A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch
  ☆95Updated 2 months ago
• IvanYashchuk / jax-fenics-adjoint
  Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint
  ☆91Updated last year
• pnkraemer / matfree
  Matrix-free linear algebra in JAX.
  ☆107Updated 2 months ago
• massastrello / awesome-implicit-neural-models
  ☆98Updated 3 years ago
• tychovdo / lila
  Code for "Invariance Learning in Deep Neural Networks with Differentiable Laplace Approximations"
  ☆23Updated 2 years ago
• patrick-kidger / FasterNeuralDiffEq
  Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)
  ☆86Updated 2 years ago
• afqueiruga / StatefulOdeNets
  Refining continuous-in-depth neural networks
  ☆39Updated 3 years ago
• dimitra-maoutsa / odes_for_sdes
  Deterministic particle dynamics for simulating Fokker-Planck probability flows
  ☆24Updated last year
• katalinic / sdflows
  JAX exponential map normalising flows on sphere
  ☆17Updated 4 years ago
• ASEM000 / pytreeclass
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆41Updated last month
• ucl-bug / jaxdf
  A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
  ☆122Updated 2 months ago
• facebookresearch / neural-conservation-law
  code for "Neural Conservation Laws A Divergence-Free Perspective".
  ☆35Updated last year
• nimadehmamy / L-conv-code
  Lie Algebra Convolutional Network implementation
  ☆42Updated 2 years ago
• martenlienen / finite-element-networks
  Reference implementation of Finite Element Networks as proposed in "Learning the Dynamics of Physical Systems from Sparse Observations wi…
  ☆69Updated 6 months ago
• SymJAX / SymJAX
  Documentation:
  ☆120Updated last year
• elizabethnewman / hessQuik
  Computing gradients and Hessians of feed-forward networks with GPU acceleration
  ☆19Updated 9 months ago
• zalandoresearch / JaxRK
  RKHS feature vectors, operators, and statistical models using JAX for automatic differentiation
  ☆8Updated 3 years ago
• glouppe / flowing-with-jax
  ☆15Updated 3 years ago
• CW-Huang / CP-Flow
  Convex potential flows
  ☆78Updated 3 years ago
• akbir / deq-jax
  [NeurIPS'19] Deep Equilibrium Models Jax Implementation
  ☆38Updated 4 years ago
• CUAI / Neural-Manifold-Ordinary-Differential-Equations
  [NeurIPS 2020] Neural Manifold Ordinary Differential Equations (https://arxiv.org/abs/2006.10254)
  ☆115Updated last year
• f-dangel / vivit
  [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…
  ☆17Updated last year