Add initial feed-forward (no-history) model, quantized sin experiments

.gitignore

@@ -127,3 +127,10 @@ dmypy.json
 
 # Pyre type checker
 .pyre/
+
+# Data Directories
+data/
+
+# Mac OS
+.DS_Store
+.idea

README.md

@@ -1 +1,48 @@
-# s4
+# S4
+
+## Quickstart (Development)
+
+We have two `requirements.txt` files that hold dependencies for the current project: one that is tailored to CPUs,
+the other that installs for GPU.
+
+### CPU-Only (MacOS, Linux)
+
+```bash 
+# Set up virtual/conda environment of your choosing & activate...
+pip install -r requirements-cpu.txt
+```
+
+### GPU (CUDA 11.3)
+
+```bash
+# Set up virtual/conda environment of your choosing & activate...
+pip install -r requirements.txt
+```
+
+## Dependencies from Scratch
+
+In case the above `requirements.txt` don't work, here are the commands used to download dependencies.
+
+### CPU-Only
+
+```bash
+# Set up virtual/conda environment of your choosing & activate... then install the following:
+pip install --upgrade "jax[cpu]"
+pip install torch torchvision torchaudio
+pip install flax
+
+# Defaults
+pip install ipython matplotlib
+```
+
+### GPU (CUDA > 11, CUDNN > 8.0.5)
+
+```bash
+# Set up virtual/conda environment of your choosing & activate... then install the following:
+pip install jax[cuda11_cudnn805] -f https://storage.googleapis.com/jax-releases/jax_releases.html
+pip install torch torchvision torchaudio
+pip install flax
+
+# Defaults
+pip install ipython matplotlib 
+```

requirements-cpu.txt

@@ -0,0 +1,47 @@
+absl-py==1.0.0
+appnope==0.1.2
+backcall==0.2.0
+certifi==2021.10.8
+chex==0.1.0
+click==8.0.3
+cycler==0.11.0
+decorator==5.1.0
+dm-tree==0.1.6
+flatbuffers==2.0
+flax==0.3.6
+fonttools==4.28.5
+ipython==7.30.1
+jax==0.2.26
+jaxlib==0.1.75
+jedi==0.18.1
+kiwisolver==1.3.2
+matplotlib==3.5.1
+matplotlib-inline==0.1.3
+msgpack==1.0.3
+mypy-extensions==0.4.3
+numpy==1.21.5
+opt-einsum==3.3.0
+optax==0.1.0
+packaging==21.3
+parso==0.8.3
+pathspec==0.9.0
+pexpect==4.8.0
+pickleshare==0.7.5
+Pillow==8.4.0
+platformdirs==2.4.0
+prompt-toolkit==3.0.24
+ptyprocess==0.7.0
+Pygments==2.10.0
+pyparsing==3.0.6
+python-dateutil==2.8.2
+scipy==1.7.3
+six==1.16.0
+tomli==1.2.3
+toolz==0.11.2
+torch==1.10.1
+torchaudio==0.10.1
+torchvision==0.11.2
+tqdm==4.62.3
+traitlets==5.1.1
+typing-extensions==3.10.0.2
+wcwidth==0.2.5

requirements.txt

@@ -0,0 +1,54 @@
+absl-py==1.0.0
+backcall==0.2.0
+certifi==2021.10.8
+chex==0.1.0
+configparser==5.0.2
+cycler==0.11.0
+decorator==5.1.0
+dm-tree==0.1.6
+docker-pycreds==0.4.0
+flatbuffers==2.0
+flax==0.3.6
+fonttools==4.28.5
+gitdb==4.0.7
+GitPython==3.1.18
+ipython==7.30.1
+jax==0.2.26
+jaxlib==0.1.75+cuda11.cudnn805
+jedi==0.18.1
+kiwisolver==1.3.2
+matplotlib==3.5.1
+matplotlib-inline==0.1.3
+msgpack==1.0.3
+numpy==1.21.5
+opencv-python==4.5.1.48
+opt-einsum==3.3.0
+optax==0.1.0
+packaging==21.3
+parso==0.8.3
+pathtools==0.1.2
+pexpect==4.8.0
+pickleshare==0.7.5
+Pillow==8.4.0
+promise==2.3
+prompt-toolkit==3.0.24
+psutil==5.8.0
+ptyprocess==0.7.0
+Pygments==2.10.0
+pyparsing==3.0.6
+python-dateutil==2.8.1
+scipy==1.7.3
+sentry-sdk==1.1.0
+shortuuid==1.0.1
+six==1.16.0
+smmap==4.0.0
+subprocess32==3.5.4
+toolz==0.11.2
+torch==1.10.1
+torchaudio==0.10.1
+torchvision==0.11.2
+tqdm==4.62.3
+traitlets==5.1.1
+typing-extensions==3.10.0.2
+wandb==0.10.32
+wcwidth==0.2.5