JAX @ Google
Check out my website at robertdyro.com
JAX @ Google
Check out my website at robertdyro.com
Friendly Interface to JAX, `jaxfi` simplifies JAX interface to replicate PyTorch's ease of use.
Python 3
JAX library for first- and second-order optimization sensitivity analysis
PyTorch library for first- and second-order optimization sensitivity analysis
High-level Python Particle Sequential Convex Programming Model Predictive Control (SCP PMPC) interface
Jupyter Notebook 15
Day of Week | April Apr | May May | June Jun | July Jul | August Aug | September Sep | October Oct | November Nov | December Dec | January Jan | February Feb | March Mar | April Apr | ||||||||||||||||||||||||||||||||||||||||
Sunday Sun | |||||||||||||||||||||||||||||||||||||||||||||||||||||
Monday Mon | |||||||||||||||||||||||||||||||||||||||||||||||||||||
Tuesday Tue | |||||||||||||||||||||||||||||||||||||||||||||||||||||
Wednesday Wed | |||||||||||||||||||||||||||||||||||||||||||||||||||||
Thursday Thu | |||||||||||||||||||||||||||||||||||||||||||||||||||||
Friday Fri | |||||||||||||||||||||||||||||||||||||||||||||||||||||
Saturday Sat |
The mask in jvp might not match the pytree structure of grad_y
.
Unfortunately, I don't have an open-source repro at the moment, but these lines ass…