From 00cd5b38f26e40fbb0dd21e9682e9a9580fe3e78 Mon Sep 17 00:00:00 2001 From: y-jan137 Date: Wed, 18 Mar 2026 09:56:29 +0300 Subject: Reformat --- include/adam.h | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) create mode 100644 include/adam.h (limited to 'include/adam.h') diff --git a/include/adam.h b/include/adam.h new file mode 100644 index 0000000..37f5767 --- /dev/null +++ b/include/adam.h @@ -0,0 +1,16 @@ +#pragma once + +typedef struct { + double *m; + double *v; + int nparams; + double lr; + double beta1; + double beta2; + double eps; + int t; +} Adam; + +Adam adam_init(int nparams, double lr, double beta1, double beta2, double eps); +void adam_update(Adam *a, double *theta, const double *grad); +void adam_free(Adam *a); -- cgit v1.2.3