Open source code for AlphaFold 2.

Parsing tools for GTF (gene transfer format) files

Fast and memory-efficient exact attention

Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch

Technical Analysis Library using Pandas and Numpy

Robust representation of semantically constrained graphs, in particular for molecules in chemistry

A modular framework for neural networks with Euclidean symmetry

Hierarchical Generation of Molecular Graphs using Structural Motifs

Graphormer is a general-purpose deep learning backbone for molecular modeling.

Structured state space sequence models

A Tree based feature selection tool which combines both the Boruta feature selection algorithm with shapley values.

PyHessian is a Pytorch library for second-order based analysis and training of Neural Networks

Scalable, Portable and Distributed Gradient Boosting (GBDT, GBRT or GBM) Library, for Python, R, Java, Scala, C++ and more. Runs on single machine, Hadoop, Spark, Dask, Flink and DataFlow

PyTorch implementation for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)

Official code for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)

High-fidelity performance metrics for generative models in PyTorch

Quasi Hyperbolic Rectified DEMON Adam/Amsgrad with AdaMod, Gradient Centralization, Lookahead, iterative averaging and decorrelated Weight Decay

Code to accompany the paper Radial Bayesian Neural Networks: Beyond Discrete Support In Large-Scale Bayesian Deep Learning

A curated list of practical financial machine learning tools and applications.

The official PyTorch implementation for NCSNv2 (NeurIPS 2020)

Noise Conditional Score Networks (NeurIPS 2019, Oral)

Quantum Hamiltonian-Based Models built on TensorFlow Quantum

Pytorch port of Google Research's LEAF Audio paper

torch-optimizer -- collection of optimizers for Pytorch

Is the attention layer even necessary? (https://arxiv.org/abs/2105.02723)

LEAF is a learnable alternative to audio features such as mel-filterbanks, that can be initialized as an approximation of mel-filterbanks, and then be trained for the task at hand, while using a ve…

Mastering Atari with Discrete World Models

A library for programmatically generating equivariant layers through constraint solving

Path-SGD: Path-Normalized Optimization in Deep Neural Networks

Pytorch library for fast transformer implementations