Michael Poli's repositories
awesome-neural-ode
A collection of resources regarding the interplay between differential equations, deep learning, dynamical systems, control and numerical methods.
wattnet-fx-trading
WATTNet: Learning to Trade FX with Hierarchical Spatio-Temporal Representations of Highly Multivariate Time Series
awesome-latent-control
A collection of resources regarding the interplay between control, causality, latent structure and self-supervision.
accelerate
🚀 A simple way to train and use PyTorch models with multi-GPU, TPU, mixed-precision
AutoAWQ
AutoAWQ implements the AWQ algorithm for 4-bit quantization with a 2x speedup during inference.
bayesian-sde
Code for "Infinitely Deep Bayesian Neural Networks with Stochastic Differential Equations"
cheatsheets
Official Matplotlib cheat sheets
DeepSpeed
DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.
flash-attention
Fast and memory-efficient exact attention
gdown
Download a large file from Google Drive (curl/wget fails because of the security notice).
helm-safari
Holistic Evaluation of Language Models (HELM), a framework to increase the transparency of language models (https://arxiv.org/abs/2211.09110).
metrics
Machine learning metrics for distributed, scalable PyTorch applications.
PythonProgrammingPuzzles
A Dataset of Python Challenges for AI Research
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
pytorch-lightning
The lightweight PyTorch wrapper for ML researchers. Scale your models. Write less boilerplate
pytorch-lightning-bolts
Toolbox of models, callbacks, and datasets for AI/ML researchers.
score_sde_pytorch
PyTorch implementation for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)
torchdiffeq
Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.
zymrael.github.io
A beautiful, simple, clean, and responsive Jekyll theme for academics