Skip to content

neilkichler/diff-ml

Repository files navigation

Second-Order Differential Machine Learning

For detailed background information, consider looking into:

https://neilkichler.github.io/master-thesis/Thesis.pdf

Notebooks

The notebooks contain the majority of the code right now and accumulate the ideas that are needed for the proposed methods. The package currently implements the core functions needed for Sobolev Training / Differential Machine Learning.

Installation

Clone the repo and execute the following inside the root folder.

python -m pip install -e .

Requires Python 3.9+, JAX 0.4.16+ and Equinox 0.10.5+.

Development

We use Hatch as the project manager. The usual commands apply.

Show all available scripts

hatch env show

Run case study

hatch -e example run python examples/bachelier/bachelier.py

Run Tests

hatch run test:test

Build project wheel

hatch build

Lint project

hatch run lint:fmt