fff-rs / juice

The Hacker's Machine Learning Engine

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Native feature parity items

subversive-owl opened this issue · comments

I'm curious what the current list of items needed to catch native up to the current feature set is, across all the repositories. (My ulterior motive here is not having CUDA-compatible hardware right now.)

This would help provide a reference implementation on vanilla machines that can be used to check against the OpenCL or CUDA implementations in case hardware or framework bugs show up there.

I think there is only one thing that really lacks native impl - backpropagation for convolution for native for arbitrary dimensions, although one could split up those dimensions to simplify / split the task

ok, thanks! i'll take a stab at it and see if i can get the existing juice-examples running natively.

Any progress, anything I can help with?

Testing right now

it's very slow on my machine, but mnist is running!

Slow execution speeds were expected, but yay to actually having working code 👍

WIP calls at: fff-rs/juice-examples#19

Ticket for coaster-nn: fff-rs/coaster-nn#8