#🚀 Introducing EDUX: A Fresh Java ML Library 🚀 - We need YOU!

1 messages · Page 1 of 1 (latest)

opaque hill
#

Does your MultilayerPerceptron support LSTMs and similar?
or networks where different layers have a different number of units/neurons?

#

Also, is training/inference on the GPU supported? If so, which GPUs can be used? Only CUDA or also other GPUs?

unreal ravine
# opaque hill Does your `MultilayerPerceptron` support LSTMs and similar? or networks where di...

Actually we have an early prototype for Nivida CUDA 12.X. Full release of this feature is scheduled for Release 1.0.6. https://github.com/Samyssmile/edux/issues/36 If you have knowledge for this, you are welcome to work on this ticket.

GitHub
NVIDIA CUDA Support · Issue #36 · Samyssmile/edux

Description: As our library grows and tackles more complex ML tasks, the need for leveraging the power of GPU becomes vital to ensure optimal performance and resource utilization. The goal of this ...

unreal ravine