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
- Hessian spectral density estimation in TF and Jax☆120Updated 4 years ago
- Reinforcement learning library in JAX.☆102Updated last year
- Convolutional Neural Tangent Kernel☆108Updated 5 years ago
- CHOP: An optimization library based on PyTorch, with applications to adversarial examples and structured neural network training.☆77Updated 10 months ago
- paper lists and information on mean-field theory of deep learning☆75Updated 5 years ago
- ☆80Updated 3 years ago
- ☆153Updated last year
- DeepOBS: A Deep Learning Optimizer Benchmark Suite☆103Updated last year
- Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"☆102Updated 4 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆106Updated 2 years ago
- Parameter-Free Optimizers for Pytorch☆109Updated 8 months ago
- Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)☆58Updated 3 years ago
- ☆163Updated 5 months ago
- This repository contains a simple implementation of Interval Bound Propagation (IBP) using TensorFlow: https://arxiv.org/abs/1810.12715☆156Updated 5 years ago
- ☆17Updated 5 years ago
- PEPit is a package enabling computer-assisted worst-case analyses of first-order optimization methods.☆83Updated 3 months ago
- A collection of meta-learning algorithms in Jax☆23Updated 2 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆57Updated last year
- ☆98Updated 3 years ago
- pyhessian is a TensorFlow module which can be used to estimate Hessian matrices☆24Updated 3 years ago
- Evaluating Robustness of Neural Networks with Mixed Integer Programming☆114Updated last month
- Loopy belief propagation for factor graphs on discrete variables in JAX☆137Updated 3 months ago
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆60Updated 2 years ago
- ☆156Updated 2 years ago
- Annotating tensor shapes using Python types☆158Updated last year
- Code for NeurIPS 2019 paper: "Tensor Programs I: Wide Feedforward or Recurrent Neural Networks of Any Architecture are Gaussian Processes…☆242Updated 4 years ago
- simple JAX-/NumPy-based implementations of NGD with exact/approximate Fisher Information Matrix both in parameter-space and function-spac…☆14Updated 4 years ago
- ☆33Updated 4 years ago
- ☆50Updated last year