froystig

ML researcher, JAX co-author

@google-deepmindsfba

Pinned Repositories

jax
my jax fork
Language:Python1 0 00
jaxopt
Hardware accelerated, batchable and differentiable optimizers in JAX.
Language:Python0 0 00
rust
a safe, concurrent, practical language
Language:Rust0 1 00
scaling-book
my scaling-book fork
Language:HTML00
trajax
Language:Python0 0 00
vowpal_wabbit
Vowpal Wabbit is a machine learning system which pushes the frontier of machine learning with techniques such as online, hashing, allreduce, reductions, learning2search, active, and interactive learning.
Language:C++0 1 00
jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Language:Python31.7k 333 6k3k
scaling-book
Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs
Language:HTML219 7 431
xla
A machine learning compiler for GPUs, CPUs, and ML accelerators
Language:C++3k 45 422520
rust
Empowering everyone to build reliable and efficient software.
Language:Rust102k 1.5k 57.1k13.2k

froystig's Repositories

froystig/jax
my jax fork
Language:Python1 0 00
froystig/jaxopt
Hardware accelerated, batchable and differentiable optimizers in JAX.
Language:Python0 0 00
froystig/rust
a safe, concurrent, practical language
Language:Rust0 1 00
froystig/scaling-book
my scaling-book fork
Language:HTML00
froystig/trajax
Language:Python0 0 00
froystig/vowpal_wabbit
Vowpal Wabbit is a machine learning system which pushes the frontier of machine learning with techniques such as online, hashing, allreduce, reductions, learning2search, active, and interactive learning.
Language:C++0 1 00