Lotta work here

This commit is contained in:
Tim O\'Neil committed 2025-08-27 14:02:03 -07:00
1 parent d83d07a823
commit d89095e49b
34 files changed
+3207 -186

No files matched your search

+20
View File
@@ -0,0 +1,20 @@
#pragma once
#include "../core/tensor.hpp"
#include <vector>
namespace lm {
class AdamOptimizer {
public:
AdamOptimizer(float learning_rate = 0.001, float beta1 = 0.9, float beta2 = 0.999, float epsilon = 1e-8);
void step(std::vector<Tensor>& parameters); // Remove const
void zero_grad(std::vector<Tensor>& parameters); // Remove const
private:
float learning_rate_, beta1_, beta2_, epsilon_;
int timestep_;
std::vector<Tensor> m_, v_; // First and second moment estimates
};
} // namespace lm