ddrous / parajax

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

graphpint

Scaling differential equations via transferable dynamics. We focus on generalising dysnmical systems to new environments. We treat a each of those environments as a node, and the edges carry the difference in network weight norms between them.

Features

  • Faster discovery of dynamical systems.
  • Better learning with advanced regularisation of the dynamics.
  • Transfer learning with all its advantages (sim2real, etc...).
  • Interpretable representations of high dimensional data via latent variable models.

Getting started

pip install graphpint

ToDos

  • Build single examples script for neural ODE, APHYNITY, etc.
  • Build a unifying framework

Flowchart

Dependencies

  • JAX
  • Equinox
  • Jraph

About


Languages

Language:Python 95.3%Language:Shell 4.7%