Skip to content

Use Flux.jl for models#10

Open
pxl-th wants to merge 5 commits intomainfrom
pxl-th/flux
Open

Use Flux.jl for models#10
pxl-th wants to merge 5 commits intomainfrom
pxl-th/flux

Conversation

@pxl-th
Copy link
Member

@pxl-th pxl-th commented Jun 2, 2023

  • Use Flux.jl for MLP models, since it now supports AMDGPU.jl.
  • Remove in-house MLP implementation.
  • Remove sync_free!, AMDGPU (pxl-th/hiprtc branch) now has stream-ordered allocator.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant