🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
-
Updated
May 16, 2024 - Python
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.
Orbax provides common utility libraries for JAX users.
Amos optimizer with JEstimator lib.
Flax Engine – multi-platform 3D game engine
Tevatron - A flexible toolkit for neural retrieval research and development.
Code I used for my YouTube videos
EasyDeL is an OpenSource Library to make your training faster and more Optimized With cool Options for training and serving Both in Python And Mojo🔥
Deep Learning examples using the Jax ecosystem of libraries
An experimental code base for system identification 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).
A Jax-based library for designing and training transformer models from scratch.
Jax, Flax, examples (ImageClassification, SemanticSegmentation, and more...)
Flax Engine Documentation
Official code for Continuous-Time Functional Diffusion Processes (NeurIPS 2023).
Lightning-like training API for JAX with Flax
Generative neural networks for 3D terrain.
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."