seqax icon indicating copy to clipboard operation
seqax copied to clipboard

verified publisher icon MatX-inc
→

Metadata

seqax = sequence modeling + JAX

Results 1 seqax issues
Sort by recently updated
recently updated
newest added

CUDA and Jax version

1 comment

Hi, Thank you so much for this repo! What cuda version were you using when you trained with GPUs and did you use a specific Jax version as well? Thank...

cerabinowitz avatar cerabinowitz

← Metadata

130
Stars
10
Forks
Watchers

Owner

verified publisher icon MatX-inc

Metadata

seqax = sequence modeling + JAX

Back