-
Updated
Feb 11, 2021 - C#
flax
Here are 210 public repositories matching this topic...
Flax Engine Documentation compiled for static hosting
-
Updated
May 22, 2024 - HTML
JAX-based Model Explanation and Interpretation Library
-
Updated
Apr 10, 2024 - Jupyter Notebook
Training GPT-2 on FineWeb-Edu in JAX/Flax
-
Updated
Aug 20, 2024 - Python
-
Updated
Mar 19, 2023 - Jupyter Notebook
Machine learning samples
-
Updated
Sep 23, 2024
Graphs the activity of a flax harvester in a linux terminal.
-
Updated
Sep 17, 2021 - C
Unfinished library for the NLP tasks using Jax and flax
-
Updated
Sep 10, 2022 - Python
This repository implements sentiment classifier with Google Jax using Bert transformer as backbone. It also shows model checkpointing and loading for inference.
-
Updated
Apr 27, 2023 - Python
An implementation of Legendre Memory Units in Flax/Jax.
-
Updated
Feb 29, 2024 - Python
Common practices for distributed training using various backends
-
Updated
Sep 23, 2023 - Python
A re-implementation of Callum McDougall's "Transformer from Scratch" using Flax.
-
Updated
Sep 7, 2023 - Python
Improve this page
Add a description, image, and links to the flax topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the flax topic, visit your repo's landing page and select "manage topics."