PhD student at @ucl-dark
deq-jax. [NeurIPS'19] Deep Equilibrium Models Jax Implementation
ray_tracer. Ray-tracer written in go
gymnax. RL Environments in JAX 🌍
akbir.github.io. Personal Website