google-deepmind / jax_verify
Neural network verification in JAX
☆141Updated last year
Alternatives and similar repositories for jax_verify:
Users that are interested in jax_verify are comparing it to the libraries listed below
- Methods to soundly verify deep neural networks☆228Updated last year
- CHOP: An optimization library based on PyTorch, with applications to adversarial examples and structured neural network training.☆77Updated last year
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆185Updated 3 months ago
- Hessian spectral density estimation in TF and Jax☆122Updated 4 years ago
- Reinforcement learning library in JAX.☆100Updated last year
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆60Updated 3 years ago
- Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)☆61Updated 3 years ago
- Pytorch implementation of KFAC and E-KFAC (Natural Gradient).☆130Updated 5 years ago
- paper lists and information on mean-field theory of deep learning☆75Updated 6 years ago
- Repository containing Pytorch code for EKFAC and K-FAC perconditioners.☆141Updated last year
- Annotating tensor shapes using Python types☆158Updated last year
- DeepOBS: A Deep Learning Optimizer Benchmark Suite☆104Updated last year
- 🧀 Pytorch code for the Fromage optimiser.☆123Updated 8 months ago
- ☆154Updated last year
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆108Updated 2 years ago
- Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"☆105Updated 4 years ago
- ☆157Updated 2 years ago
- PyTorch-SSO: Scalable Second-Order methods in PyTorch☆145Updated last year
- ☆166Updated 7 months ago
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆112Updated this week
- ☆68Updated last year
- Convolutional Neural Tangent Kernel☆109Updated 5 years ago
- This repository contains a simple implementation of Interval Bound Propagation (IBP) using TensorFlow: https://arxiv.org/abs/1810.12715☆156Updated 5 years ago
- Evaluating Robustness of Neural Networks with Mixed Integer Programming☆115Updated last month
- LipSDP - Lipschitz Estimation for Neural Networks☆65Updated 3 years ago
- This repository contains the Julia code for the paper "Competitive Gradient Descent"☆23Updated 5 years ago
- A collection of meta-learning algorithms in Jax☆22Updated 2 years ago
- ☆80Updated 3 years ago
- Experiment code for "Randomized Automatic Differentiation"☆66Updated 4 years ago
- pyhessian is a TensorFlow module which can be used to estimate Hessian matrices☆24Updated 4 years ago