yuehhua / Hopfields.jl

Modern Hopfield layer implementations in Julia

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Hopfields.jl

Stable Dev Build Status Coverage

Hopfields.jl provides basic implementation of Hopfield layers that can be stacked in deep learning model. It is implemented based on Flux.jl framework and is CUDA-available with CUDA.jl.

References

This repository contains the Julia implementation of paper Hopfield Networks is All You Need. For more details, you can check their blog post and repository ml-jku/hopfield-layers.

About

Modern Hopfield layer implementations in Julia

License:MIT License


Languages

Language:Julia 100.0%