Skip to content

Instantly share code, notes, and snippets.

View ghubnerr's full-sized avatar
🍀
Probably crying to a Rust file

Gabriel Lucchesi ghubnerr

🍀
Probably crying to a Rust file
View GitHub Profile
@ghubnerr
ghubnerr / vae_mnist_jax.ipynb
Created April 8, 2025 17:18
vae_mnist_jax.ipynb
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@ghubnerr
ghubnerr / flax-cnns-rnns-lstms.ipynb
Created April 8, 2025 17:11
flax-cnns-rnns-lstms.ipynb
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@ghubnerr
ghubnerr / intro-to-gradient-descent.ipynb
Last active March 8, 2025 01:16
intro-to-gradient-descent.ipynb
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@ghubnerr
ghubnerr / jvps-and-vjps.ipynb
Last active March 17, 2025 03:09
jvps-and-vjps.ipynb
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@ghubnerr
ghubnerr / gpt2-kv-cache-kernel-fusions.ipynb
Created March 7, 2025 19:27
GPT2 (+KV-Cache & Kernel Fusions).ipynb
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@ghubnerr
ghubnerr / vae-mnist-jax.ipynb
Created January 20, 2025 21:22
vae-mnist-jax.ipynb
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@ghubnerr
ghubnerr / flax-stuff.ipynb
Created December 21, 2024 22:45
FLAX-stuff.ipynb
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@ghubnerr
ghubnerr / jax-mlp.ipynb
Created December 21, 2024 21:07
JAX-MLP.ipynb
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
gcloud artifacts repositories create my-web-app --repository-format=docker --location=us-west2 --description="Repo for section a images."
gcloud builds submit --region=us-west2 --config cloudbuild.yaml
@ghubnerr
ghubnerr / dqn.py
Created October 6, 2023 03:16
AtariBreakout
import numpy as np
import tensorflow as tf
from tensorflow import keras
from keras.optimizers.legacy import Adam
class ReplayBuffer():
def __init__(self, max_size, input_dims):
self.mem_size = max_size
self.mem_cntr = 0