Robert Dyro

torch2jax. Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.

mpcjax. A JAX library for Sequential Convex Programming (SCP) Model Predictive Control (MPC) path planning

Pallas-Puzzles. Jupyter Notebook

tune-jax. Microbenchmarking hyperparameter tuning for JAX functions.

SpAutoDiff.jl. Full [Sp]arse Graph [AutoDiff]erentiation Package for Julia

jaxfi-JAXFriendlyInterface. Friendly Interface to JAX, `jaxfi` simplifies JAX interface to replicate PyTorch's ease of use.

gpu_ragged_dot. Ragged dot op with auto-diff in JAX Pallas

CUDA_QP_Solver. A small projection-based QP solver for a single CUDA thread

sensitivity_jax. JAX library for first- and second-order optimization sensitivity analysis

JAX-and-NNX-Tour. Jupyter Notebook

weather_landscape. Visualizing Weather Forecasts Through Landscape Imagery

libtpuinfo. Go

jax-triton. jax-triton contains integrations between JAX and OpenAI Triton

avg_turtlebot. Python

nvtop-with-tpu. GPU & Accelerator process monitoring for AMD, Apple, Huawei, Intel, NVIDIA and Qualcomm

sensitivity_torch. PyTorch library for first- and second-order optimization sensitivity analysis

skip_trajectory_via_scp_mpc. Julia

cvar-rl. Python

Mosek_jll.jl. Julia

multipathpp. Cleaned up repository with multipath-plus-plus from stepankonev

NVIDIA-GPUs-Info-and-Used-Price. A snapshot CSV file containing the stats, used prices and Geekbench CUDA scores for NVIDIA GPUs.

guest_container_tool. This basic tool allows you to give others access to your computational resources inside an isolated container, without access to your server or other containers.