Tevatron - A flexible toolkit for neural retrieval research and development.
-
Updated
May 29, 2024 - Python
Tevatron - A flexible toolkit for neural retrieval research and development.
🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
NAACL '24 (Demo) / MlSys @ NeurIPS '23 - RedCoast: A Lightweight Tool to Automate Distributed Training and Inference
Orbax provides common utility libraries for JAX users.
Accelerate your training with this open-source library. Optimize performance with streamlined training and serving options with JAX. 🚀
Flax Engine – multi-platform 3D game engine
Clean single-file implementation of offline RL algorithms in JAX
Code I used for my YouTube videos
A Jax-based library for designing and training transformer models from scratch.
Generative neural networks for 3D terrain.
Deep Learning examples using the Jax ecosystem of libraries
Flax Engine Documentation
Original Implementation of Prompt Tuning from Lester, et al, 2021
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 Documentation compiled for static hosting
Collection of example projects for Flax Engine
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."