Owen Dugan's repositories
QuantFluNNAnalysis
Data processing and neural network training for the QuantifiedFlu project
state-spaces
Sequence Modeling with Structured State Spaces
OccamNet_Versions
A compilation of all OccamNet versions
diffrax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
Gym-Snake
An OpenAI gym environment made for RL
jax-flows
Normalizing Flows in JAX 🌊
levanter
Legibile, Scalable, Reproducible Foundation Models with Named Tensors and Jax
mixed_autonomy_intersections
[ITSC 2021] Reinforcement Learning for Mixed Autonomy Intersections
nanoGPT
The simplest, fastest repository for training/finetuning medium-sized GPTs.
netket
Machine learning algorithms for many-body quantum systems
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
qst-cgan
Quantum state tomography with conditional generative adversarial networks
qst-nn
Classification and reconstruction of optical quantum states with deep neural networks
RWKV-LM
RWKV is an RNN with transformer-level LLM performance. It can be directly trained like a GPT (parallelizable). So it's combining the best of RNN and transformer - great performance, fast inference, saves VRAM, fast training, "infinite" ctx_len, and free sentence embedding.
v202
Proceedings of ICML 2023