How to use the Flax Linen API to build a convolutional neural network model and train it for image classification (using TensorFlow Datasets).
☆25Aug 16, 2023Updated 2 years ago
Alternatives and similar repositories for JAX-Flax-Tutorial-Image-Classification-with-Linen
Users that are interested in JAX-Flax-Tutorial-Image-Classification-with-Linen are comparing it to the libraries listed below
Sorting:
- This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.☆14Dec 21, 2021Updated 4 years ago
- DiffPhase: Generative Diffusion-based STFT Phase Retrieval☆16Sep 21, 2023Updated 2 years ago
- Simple, extensible implementations of some meta-learning algorithms in Jax☆11Oct 6, 2020Updated 5 years ago
- An easy way to start a python programming environment using GitHub Codespaces.☆15Sep 9, 2020Updated 5 years ago
- A Jax based meta learning library☆16Sep 15, 2022Updated 3 years ago
- Neural Networks for JAX☆84Sep 24, 2024Updated last year
- TorrentPier. Alternative compiled announcer (Ocelot)☆23Jul 12, 2024Updated last year
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆46Jan 11, 2025Updated last year
- A JAX nn library☆21Sep 9, 2025Updated 5 months ago
- ☆22Dec 15, 2023Updated 2 years ago
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.☆21Sep 18, 2020Updated 5 years ago
- ☆24Jul 31, 2023Updated 2 years ago
- A Python package of computer vision models for the Equinox ecosystem.☆111Jul 19, 2024Updated last year
- Raku library for the Mustache template format☆23Oct 11, 2022Updated 3 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Feb 26, 2023Updated 3 years ago
- Unofficial JAX implementations of deep learning research papers☆161Jun 25, 2022Updated 3 years ago
- FID computation in Jax/Flax.☆29Jul 17, 2024Updated last year
- fast.ai APL study group notes☆29Oct 19, 2022Updated 3 years ago
- ☆28Nov 18, 2022Updated 3 years ago
- Pytrees + dataclasses ❤️☆75Dec 19, 2025Updated 2 months ago
- For experiments involving instruct gpt. Currently used for documenting open research questions.☆71Nov 8, 2022Updated 3 years ago
- ☆35Apr 12, 2024Updated last year
- An unofficial implementation of the Infini-gram model proposed by Liu et al. (2024)☆33Jun 19, 2024Updated last year
- Exploring Moving Mnist dataset with forecasting algorithms☆31Apr 12, 2023Updated 2 years ago
- A Pytree Module system for Deep Learning in JAX☆213Feb 26, 2023Updated 3 years ago
- A python algorithm to change the pitch of the voice in real time☆13Dec 13, 2020Updated 5 years ago
- 💧 Inland water systems are essential to our environment because they are vital ecosystems that are bio-diverse. Thus, finding innovative…☆11Apr 26, 2020Updated 5 years ago
- CSE201 Objected-Oriented Programming in C++: Teach an AI to produce pieces of music☆12Jan 23, 2019Updated 7 years ago
- [ICML 2024] Official Repository for the paper "Transformers Get Stable: An End-to-End Signal Propagation Theory for Language Models"☆10Jul 19, 2024Updated last year
- ☆12Jan 10, 2026Updated last month
- Using JAX to generate piano music as MIDI☆39Nov 28, 2023Updated 2 years ago
- A metrics library for the JAX ecosystem☆40Mar 16, 2023Updated 2 years ago
- Texture mapping with variational auto-encoders☆40Oct 1, 2021Updated 4 years ago
- ☆44Jun 19, 2024Updated last year
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆695Jan 26, 2026Updated last month
- Implementing the Denoising Diffusion Probabilistic Model in Flax☆159Nov 1, 2022Updated 3 years ago
- ☆12Sep 28, 2024Updated last year
- Research sources on graph-based anomaly detection☆13Nov 29, 2022Updated 3 years ago
- ☆11Dec 13, 2024Updated last year