🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
-
Updated
May 22, 2024 - Python
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
Orbax provides common utility libraries for JAX users.
🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.
Tevatron - A flexible toolkit for neural retrieval research and development.
Accelerate your training with this open-source library. Optimize performance with streamlined training and serving options with JAX. 🚀
This is a JAX/Flax-based transformer language model trained on a Japanese dataset. It is based on the official Flax example code (lm1b).
Flax Engine – multi-platform 3D game engine
Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.
Deep Learning examples using the Jax ecosystem of libraries
dataset, train, inference
Jax, Flax, examples (ImageClassification, SemanticSegmentation, and more...)
An experimental code base for system identification with Jax.
Yet another RL library built with FLAX modules
Code I used for my YouTube videos
Flax Engine Documentation
Amos optimizer with JEstimator lib.
Original Implementation of Prompt Tuning from Lester, et al, 2021
Add a description, image, and links to the flax topic page so that developers can more easily learn about it.
To associate your repository with the flax topic, visit your repo's landing page and select "manage topics."