Jaroslav Sevcik's repositories
jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Language:PythonApache-2.0000
mujoco
Multi-Joint dynamics with Contact. A general purpose physics simulator.
Language:C++Apache-2.0000
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.
Language:PythonApache-2.0000
Language:PythonApache-2.0000
source-map-spec
The specification of the source map format
Language:BikeshedNOASSERTION000
xla
A machine learning compiler for GPUs, CPUs, and ML accelerators
Language:C++Apache-2.0000