Ben Wang

icon indicating a website link benwang.dev icon indicating an email [email protected]

Results 7 repositories owned by Ben Wang

mesh-transformer-jax

6.2k
Stars
887
Forks
80
Watchers

Model parallel transformers in JAX and Haiku

verified publisher icon kingoflolz

cc_img_dl

27
Stars
3
Forks
Watchers

verified publisher icon kingoflolz

CLIP_JAX

134
Stars
20
Forks
Watchers

Contrastive Language-Image Pretraining

verified publisher icon kingoflolz

fiducial

17
Stars
4
Forks
Watchers

A LFTag and TopoTag implementation in Rust

verified publisher icon kingoflolz

linear-actuator-firmware

24
Stars
4
Forks
Watchers

verified publisher icon kingoflolz

linear-actuator-hardware

48
Stars
7
Forks
Watchers

verified publisher icon kingoflolz

swarm-jax

229
Stars
21
Forks
Watchers

Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes

verified publisher icon kingoflolz