DSSC-projects / veni

Related projects: ⓘ

• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆34Updated last month
• IvanYashchuk / jax-fenics
  Differentiable interface to FEniCS for JAX
  ☆49Updated 3 years ago
• NeilGirdhar / tjax
  Tools for JAX
  ☆44Updated last week
• ucl-bug / jaxdf
  A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
  ☆119Updated this week
• ASEM000 / pytreeclass
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆39Updated 11 months ago
• luca-heltai / numerical-analysis-2021-2022
  ☆13Updated last year
• MilesCranmer / sympy2jax
  Turning SymPy expressions into JAX functions
  ☆42Updated 3 years ago
• diffprog / code
  Code for the book "The Elements of Differentiable Programming".
  ☆57Updated last month
• IvanYashchuk / jax-fenics-adjoint
  Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint
  ☆87Updated 10 months ago
• ASEM000 / kernex
  Stencil computations in JAX
  ☆66Updated 11 months ago
• tk-rusch / unicornn
  Official code for UnICORNN (ICML 2021)
  ☆27Updated 2 years ago
• pnkraemer / probdiffeq
  Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…
  ☆30Updated last month
• elizabethnewman / hessQuik
  Computing gradients and Hessians of feed-forward networks with GPU acceleration
  ☆18Updated 7 months ago
• MJHutchinson / ExtrinsicGaugeIndependentVectorGPs
  A library implementing the kernels for and experiments using extrinsic gauge equivariant vector field Gaussian Processes
  ☆25Updated 2 years ago
• ermongroup / Neural-PDE-Solver
  ☆15Updated 4 years ago
• google-research / gpax
  ☆24Updated this week
• juliagusak / neural-ode-metasolver
  Supplementary code for the paper "Meta-Solver for Neural Ordinary Differential Equations" https://arxiv.org/abs/2103.08561
  ☆25Updated 3 years ago
• atzberg / gmls-nets
  PyTorch implementation of GMLS-Nets. Machine learning methods for scattered unstructured data sets. Methods for learning differential op…
  ☆22Updated 10 months ago
• SaitejaUtpala / rieoptax
  Riemannian Optimization Using JAX
  ☆44Updated 10 months ago
• brentyi / jax_dataclasses
  Pytrees + dataclasses ❤️
  ☆60Updated 2 months ago
• francescocicala / deep-learning-with-pytorch
  Deep learning lectures I am holding for the MSc on Data Science and Scientific Computing
  ☆13Updated 2 years ago
• deep-spin / sparse_continuous_distributions
  This repository provides open-source code for sparse continuous distributions and corresponding Fenchel-Young losses.
  ☆15Updated last year
• COR-OPT / SuperPolyak.py
  Pytorch implementation of SuperPolyak subgradient method.
  ☆44Updated last year
• Romit-Maulik / PyParSVD
  An implementation of the 1. Parallel, 2. Streaming, 3. Randomized SVD using MPI4Py
  ☆57Updated 3 years ago
• liuyao12 / ConvNets-PDE-perspective
  an Open Collaborative project to explore the implications — theoretical or practical — of the PDE perspective of ConvNets
  ☆22Updated 11 months ago
• juliagusak / neural-ode-norm
  Models and code for the ICLR 2020 workshop paper "Towards Understanding Normalization in Neural ODEs"
  ☆16Updated 4 years ago
• leonard-gleyzer / connex
  Fine-grained, dynamic control of neural network topology in JAX.
  ☆21Updated last year
• papamarkou / eeyore
  MCMC methods for neural networks
  ☆10Updated 4 months ago
• crowsonkb / mdmm
  The Modified Differential Multiplier Method (MDMM) for PyTorch
  ☆50Updated 3 years ago
• tumaer / jax-sph
  JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework
  ☆48Updated 3 weeks ago