Fast and Easy Infinite Neural Networks in Python
created at April 8, 2019, 4:48 p.m.
Fourier Features Let Networks Learn High Frequency Functions in Low Dimensional Domains
created at June 16, 2020, 11:54 p.m.
The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series of tutorials (YouTube videos and Jupyter Notebooks) as well as the content I found useful while learning about the JAX ecosystem.
created at Oct. 28, 2021, 8 p.m.
JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
created at Jan. 16, 2021, 1:26 a.m.
A library for programmatically generating equivariant layers through constraint solving
created at Sept. 11, 2020, 5:40 p.m.
Nested Hierarchical Transformer https://arxiv.org/pdf/2105.12723.pdf
created at June 1, 2021, 5:09 a.m.
Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX
created at April 3, 2020, 6:52 a.m.
Functional models and algorithms for sparse signal processing
created at Dec. 22, 2020, 8:31 a.m.
🐈 Automatic differentiable spectrum modeling of exoplanets/brown dwarfs using JAX, compatible with NumPyro and JAXopt
created at Dec. 12, 2020, 3:46 a.m.
JAX + Flax implementation of "Combinatorial Optimization with Physics-Inspired Graph Neural Networks" by Schuetz et al.
created at Jan. 4, 2023, 11:09 p.m.
How to use the Flax Linen API to build a convolutional neural network model and train it for image classification (using TensorFlow Datasets).
created at Dec. 24, 2020, 3 a.m.
Inverse optimal control for continuous psychophysics
created at Aug. 24, 2021, 7:38 a.m.