Nested Hierarchical Transformer https://arxiv.org/pdf/2105.12723.pdf
updated at March 28, 2024, 10:26 a.m.
How to use the Flax Linen API to build a convolutional neural network model and train it for image classification (using TensorFlow Datasets).
updated at March 31, 2024, 6:02 p.m.
🐈 Automatic differentiable spectrum modeling of exoplanets/brown dwarfs using JAX, compatible with NumPyro and JAXopt
updated at April 27, 2024, 9:02 p.m.
Inverse optimal control for continuous psychophysics
updated at May 3, 2024, 2:34 p.m.
JAX + Flax implementation of "Combinatorial Optimization with Physics-Inspired Graph Neural Networks" by Schuetz et al.
updated at May 13, 2024, 1:51 p.m.
Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX
updated at May 18, 2024, 2:42 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.
updated at May 21, 2024, 4:43 p.m.
A library for programmatically generating equivariant layers through constraint solving
updated at May 21, 2024, 6:27 p.m.
Functional models and algorithms for sparse signal processing
updated at May 22, 2024, 12:12 p.m.
Fourier Features Let Networks Learn High Frequency Functions in Low Dimensional Domains
updated at May 24, 2024, 7:57 p.m.
JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
updated at May 26, 2024, 2:10 p.m.
Fast and Easy Infinite Neural Networks in Python
updated at May 26, 2024, 5:45 p.m.