Pinned Repositories
autodiff-simple-rs
jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
mujoco
Multi-Joint dynamics with Contact. A general purpose physics simulator.
paxml
Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimentation and parallelization, and has demonstrated industry leading model flop utilization rates.
jaro-sevcik's Repositories
jaro-sevcik/autodiff-simple-rs
jaro-sevcik/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
jaro-sevcik/mujoco
Multi-Joint dynamics with Contact. A general purpose physics simulator.
jaro-sevcik/paxml
Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimentation and parallelization, and has demonstrated industry leading model flop utilization rates.
jaro-sevcik/praxis
jaro-sevcik/source-map-spec
The specification of the source map format
jaro-sevcik/xla
A machine learning compiler for GPUs, CPUs, and ML accelerators
jaro-sevcik/xla-js