Flux.jl
GPT-2 example: add a CUDA.jl version
#2765
Merged
Overview
Commits
5
Changes
View On GitHub

GPT-2 example: add a CUDA.jl version #2765

CarloLucibello merged 5 commits into master from cl/gpt2-cuda
CarloLucibello
CarloLucibello GPT-2 example: add a CUDA.jl version next to the Reactant and PyTorch…
5b3866a1
codecov
CarloLucibello Merge branch 'master' into cl/gpt2-cuda
428aa291
CarloLucibello GPT-2 example: CUDA.jl version at micro-batch 16 with Flux 0.16.14
fe06e177
CarloLucibello GPT-2 example: link the NNlib fix for the Reactant slowdown
2e1a5bd9
CarloLucibello GPT-2 example: PyTorch with torch.compile in the comparison
40f97b46
CarloLucibello CarloLucibello marked this pull request as ready for review 4 days ago
CarloLucibello CarloLucibello merged 07ac41c7 into master 4 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum