clayton-r / symtorch

Related projects ⓘ

Alternatives and complementary repositories for symtorch

• Zymrael / torchSODE
  PyTorch block-diagonal ODE CUDA solver, designed for gradient-based optimization
  ☆16Updated 4 years ago
• patrick-kidger / sympytorch
  Turning SymPy expressions into PyTorch modules.
  ☆141Updated last year
• facebookresearch / rcpm
  Riemannian Convex Potential Maps
  ☆68Updated last year
• Information-Fusion-Lab-Umass / NuX
  Normalizing Flows using JAX
  ☆82Updated 11 months ago
• dimitra-maoutsa / odes_for_sdes
  Deterministic particle dynamics for simulating Fokker-Planck probability flows
  ☆24Updated last year
• imbirik / LieStationaryKernels
  Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"
  ☆41Updated last year
• willieneis / bayesian-algorithm-execution
  Bayesian algorithm execution (BAX)
  ☆47Updated 3 years ago
• alexOarga / haiku-geometric
  A collection of graph neural networks implementations in JAX
  ☆31Updated 11 months ago
• MilesCranmer / sympy2jax
  Turning SymPy expressions into JAX functions
  ☆42Updated 3 years ago
• akbir / deq-jax
  [NeurIPS'19] Deep Equilibrium Models Jax Implementation
  ☆37Updated 4 years ago
• SaitejaUtpala / rieoptax
  Riemannian Optimization Using JAX
  ☆45Updated last year
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆38Updated 2 weeks ago
• necludov / iMCMC
  ☆22Updated 4 years ago
• crowsonkb / mdmm
  The Modified Differential Multiplier Method (MDMM) for PyTorch
  ☆51Updated 3 years ago
• pnkraemer / matfree
  Matrix-free linear algebra in JAX.
  ☆103Updated 2 months ago
• Iolo-Jones / ManifoldDiffusionGeometry
  Code for "Manifold Diffusion Geometry: Curvature, Tangent Spaces, and Dimension"
  ☆28Updated this week
• maxxxzdn / implicit-steerable-kernels
  Official repository of Implicit Neural Convolutional Kernels for Steerable CNNs, Zhdanov et al.
  ☆26Updated 7 months ago
• nimadehmamy / L-conv-code
  Lie Algebra Convolutional Network implementation
  ☆42Updated 2 years ago
• tensorly / torch
  TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch
  ☆76Updated 5 months ago
• SymJAX / SymJAX
  Documentation:
  ☆120Updated last year
• google-deepmind / dm_hamiltonian_dynamics_suite
  ☆34Updated 3 years ago
• zalandoresearch / JaxRK
  RKHS feature vectors, operators, and statistical models using JAX for automatic differentiation
  ☆8Updated 3 years ago
• patrick-kidger / FasterNeuralDiffEq
  Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)
  ☆86Updated 2 years ago
• CUAI / Neural-Manifold-Ordinary-Differential-Equations
  [NeurIPS 2020] Neural Manifold Ordinary Differential Equations (https://arxiv.org/abs/2006.10254)
  ☆115Updated last year
• lezcano / expm
  Differentiable and numerically stable implementation of the matrix exponential
  ☆32Updated 4 years ago
• tychovdo / lila
  Code for "Invariance Learning in Deep Neural Networks with Differentiable Laplace Approximations"
  ☆23Updated last year
• NeilGirdhar / efax
  Exponential families for JAX
  ☆55Updated 3 weeks 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
  ☆120Updated last month
• MachineLearningLifeScience / stochman
  Algorithms for computations on random manifolds made easier
  ☆85Updated 11 months ago