aniketjivani

iTransformer

Official implementation for "iTransformer: Inverted Transformers Are Effective for Time Series Forecasting" (ICLR 2024 Spotlight), https://openreview.net/forum?id=JePfAI8fah

NIPS-Reproducibility-Challenge

Shape and Time Distortion Loss forTraining Deep Time Series Forecasting Models

eecs598

Advanced Topics on Systems for X

torchdiffeq

Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.

neuralgbi

Richard @rdgao & Michael @michaeldeistler: using neural network-based regression and density estimation for Generalized Bayesian Inference

mamba-chat

Mamba-Chat: A chat LLM based on the state-space model architecture 🐍

litgpt

20+ high-performance LLM implementations with recipes to pretrain, finetune and deploy at scale.

llamafile

Distribute and run LLMs with a single file.

autoreg-pde-diffusion

Benchmarking Autoregressive Conditional Diffusion Models for Turbulent Flow Simulation

Allen-Cahn-FNO

Fourier Neural Operators to solve for Allen Cahn PDE equations

fourier_neural_operator

Fourier Neural Operator

fourier_neural_operator

Use Fourier transform to learn operators in differential equations.

NeuralPointProcess

normalizing-flows

PyTorch implementation of normalizing flow models

jammy_flows

A package to describe amortized (conditional) normalizing-flow PDFs defined jointly on tensor products of manifolds with coverage control. The connection between different manifolds is fixed via an autoregressive structure.

alphafold2

To eventually become an unofficial Pytorch implementation / replication of Alphafold2, as details of the architecture get released

locationencoder

nougat

Implementation of Nougat Neural Optical Understanding for Academic Documents

cnn-surrogate

Bayesian deep convolutional encoder-decoder networks for surrogate modeling and uncertainty quantification

qgss-2023

All things Qiskit Global Summer School 2023: Theory to Implementation - Lecture notes, Labs and Solutions

EnbPI

Official code for: Conformal prediction interval for dynamic time-series (conference, ICML 21 Long Presentation) AND Conformal prediction for time-series (journal, IEEE TPAMI)

uncertainty-toolbox

Uncertainty Toolbox: a Python toolbox for predictive uncertainty quantification, calibration, metrics, and visualization

scp

R Package for Spatial Conformal Prediction

temperature_scaling

A simple way to calibrate your neural network.

Mechanical-MNIST

Mechanical-MNIST is a benchmark dataset for mechanical meta-models -- this repository contains code to generate metamodels for Mechanical-MNIST

vae_cahn-hilliard

Generative model variational autoencoder (VAE) implementation for the predictions of phase separation in binary alloys

affine-parametric-networks

Code and Data for the paper "Improving Parametric Neural Networks for High-Energy Physics (and Beyond)" MLST, at https://doi.org/10.1088/2632-2153/ac917c.

KernelsOperatorLearning

CoDeF

[CVPR 2024 Highlight] Official PyTorch implementation of CoDeF: Content Deformation Fields for Temporally Consistent Video Processing

conv_arithmetic

A technical report on convolution arithmetic in the context of deep learning