Tensors and Dynamic neural networks in Python with strong GPU acceleration

A Python-embedded modeling language for convex optimization problems.

JAX implementation of MeanFlow

Face recognition with deep neural networks.

A fast and differentiable QP solver for PyTorch.

An intelligent block matrix library for numpy, PyTorch, and beyond.

Code for minimum-entropy coupling.

Bridging deep learning and logical reasoning using a differentiable satisfiability solver.

[NeurIPS'19] Deep Equilibrium Models

Sequence modeling benchmarks and temporal convolutional networks

A method for training neural networks that are provably robust to adversarial attacks.

Tutorials and implementations for "Self-normalizing networks"

A differentiable LCP physics engine in PyTorch.

OptNet: Differentiable Optimization as a Layer in Neural Networks

[ICLR'19] Trellis Networks for Sequence Modeling

Provable adversarial robustness at ImageNet scale

PyTorch wrapper for FFTs

A Newton ADMM based solver for Cone programming.