basalt
basalt copied to clipboard
basalt-org
A Machine Learning framework from scratch in Pure Mojo 🔥
Hi, Similar to what I did for MNIST, here using the stdlib string splitting to parse the housing data. This time I checked that the loss is the same, but...
Adding the SLICE operator. - [x] forward - [ ] backward - [ ] unittests As it is right now the operator does only support slicing in one dimension at...
https://github.com/basalt-org/examples Signed-off-by: benny-nottonson
The README mentioned that more activation functions are on the road map. I have gotten started. As of publishing, I only added Threshold, but I plan to go down the...
This PR adds the code required to train a CNN on the CIFAR10 dataset. I adapted the Dataloader already there to one that looks more like a Pytorch dataloader. I...
This should not be pushed yet, its more of a rough idea to get a conversation started.
Added an implementation of MAE loss, this also adds the ABS operator (Inplace absolute value) and its associated derivative. Signed-off-by: benny-nottonson
This PR fixes a number of issues with Collection and additionally adds .reserve() and .reserve_additional() to allow Model to reserve and allocate memory at creation, rather than resizing when needed...
Metadata
Owner
Metadata
A Machine Learning framework from scratch in Pure Mojo 🔥