Add pytype ignore statements related to deprecated JAX APIs#347

copybara-service[bot] merged 1 commit into

maingoogle/learned_optimization:mainfrom

test_943879532google/learned_optimization:test_943879532Copy head branch name to clipboard