thomasgho
Follow
Stars
Graph Neural Network Library for PyTorch
Deep universal probabilistic programming with Python and PyTorch
Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.
A data augmentations library for audio, image, text, and video.
A Python framework for high performance GPU simulation and graphics
MlFinLab helps portfolio managers and traders who want to leverage the power of machine learning by providing reproducible, interpretable, and easy to use tools.
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
pip install antialiased-cnns to improve stability and accuracy
Differentiable SDE solvers with GPU support and efficient sensitivity analysis.
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
A package for the sparse identification of nonlinear dynamical systems from data
PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations
A modular framework for neural networks with Euclidean symmetry
A PyTorch and TorchDrug based deep learning library for drug pair scoring. (KDD 2022)
A curated collection of adversarial attack and defense on graph data.
Junction Tree Variational Autoencoder for Molecular Graph Generation (ICML 2018)
Vector Neurons: A General Framework for SO(3)-Equivariant Networks
Differentiable Finite Element Method with JAX
Implementation of the Equiformer, SE3/E3 equivariant attention network that reaches new SOTA, and adopted for use by EquiFold for protein folding
[ICLR 2023 Spotlight] Equiformer: Equivariant Graph Attention Transformer for 3D Atomistic Graphs
Demo Weibull Time-to-event Recurrent Neural Network in Keras
Official implementation of UTNet: A Hybrid Transformer Architecture for Medical Image Segmentation
Code for "Infinitely Deep Bayesian Neural Networks with Stochastic Differential Equations"
A short and easy PyTorch implementation of E(n) Equivariant Graph Neural Networks