6 projects
tune-jax
Microbenchmarking hyperparameter tuning for JAX functions.
wrap-torch2jax
Wrap your PyTorch for JAX! This package allows no-copy PyTorch calling from JAX under both eager execution and JIT.
jaxfi
Friendly Interface to JAX, that behaves similar to PyTorch while maintaining compatibility.
mpcjax
A pure-JAX, CPU or GPU Model Predictive Control solver.
semantic-bibtool
Command Line Tool for generating bibliography files using Semantic Scholar API
pmpc
Python-interface Particle Sequential Convex Programming Model Predictive Control (SCP PMPC) interface.