search

google-research / big_vision

Official codebase used to develop Vision Transformer, SigLIP, MLP-Mixer, LiT and more.
Apache License 2.0
2.04k stars 140 forks source link

GPU test #95

Closed prazek closed 4 months ago

prazek commented 4 months ago

Initial commit with a test for jax on GPUs

Co-authored-by: tobiaslee tobiaslee@qq.com

prazek commented 4 months ago

Apologies for opening pull request, wanted to send pull request to my fork and I have not noticed it created one to the original repo