/neuralhub

A hub for learning dynamical systems

Primary LanguagePython

Neural Hub

This repository contains building blocks for scaling deep learning models for science ! We prototype ideas here before moving the their respective repos.

Getting started

pip install neuralhub

Dependencies

  • JAX
  • Equinox
  • Jraph