Models and examples built with TensorFlow

Siamese and triplet networks with online pair/triplet mining in PyTorch

Demo for Multi-Layer ISTA and Multi-Layer FISTA algorithms for convolutional neural networks, as described in J. Sulam, A. Aberdam, A. Beck, M. Elad, (2018). On Multi-Layer Basis Pursuit, Efficient…

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

3D U-Net model for volumetric semantic segmentation written in pytorch

Fully Convolutional DenseNets for semantic segmentation.

torch-optimizer -- collection of optimizers for Pytorch

TensorDict is a pytorch dedicated tensor container.