apax-hub / apax

A flexible and performant framework for training machine learning potentials.
MIT License
15 stars 2 forks source link

allow for non NTK linear layers in model #283

Closed M-R-Schaefer closed 3 months ago