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