pmap->jit transition + misc updates accumulated over time

google-research / big_vision

Official codebase used to develop Vision Transformer, SigLIP, MLP-Mixer, LiT and more.

Apache License 2.0

2.16k stars 147 forks source link

Closed akolesnikoff closed 10 months ago