GitHub / Ceyron / trainax

Training methodologies for autoregressive neural operators in JAX.

JSON API: http://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/Ceyron%2Ftrainax
PURL: pkg:github/Ceyron/trainax

Stars: 2
Forks: 0
Open issues: 1

License: mit
Language: Python
Size: 6.43 MB
Dependencies parsed at: Pending

Created at: over 1 year ago
Updated at: 9 months ago
Pushed at: 9 months ago
Last synced at: 9 months ago

Topics: autoregressive-models, deep-learning, differentiable-physics, optax, optimization, timeseries-forecasting, unrolled-training

Loading...