ml-gde / jax-layers
A collection of reusable, high-performance, well-documented, thorough-tested layers and models in Jax
β11Updated last week
Alternatives and similar repositories for jax-layers:
Users that are interested in jax-layers are comparing it to the libraries listed below
- JAX Implementation of Black Forest Labs' Flux.1 family of modelsβ30Updated 5 months ago
- This repository shows various ways of deploying a vision model (TensorFlow) from π€ Transformers.β30Updated 2 years ago
- β18Updated 2 years ago
- A set of Python scripts that makes your experience on TPU betterβ50Updated 8 months ago
- β46Updated 4 months ago
- A library for squeakily cleaning and filtering language datasets.β46Updated last year
- ML/DL Math and Method notesβ59Updated last year
- This is a port of Mistral-7B model in JAXβ32Updated 9 months ago
- Scripts to convert datasets from various sources to Hugging Face Datasets.β57Updated 2 years ago
- Contains my experiments with the `big_vision` repo to train ViTs on ImageNet-1k.β22Updated 2 years ago
- Various handy scripts to quickly setup new Linux and Windows sandboxes, containers and WSL.β40Updated 2 weeks ago
- β24Updated last year
- Presents comprehensive benchmarks of XLA-compatible pre-trained models in Keras.β37Updated last year
- Various transformers for FSDP researchβ37Updated 2 years ago
- β31Updated this week
- Code for NeurIPS LLM Efficiency Challengeβ57Updated 11 months ago
- Genalog is an open source, cross-platform python package allowing generation of synthetic document images with custom degradations and teβ¦β42Updated last year
- Pragmatic approach to parsing import profiles for CI'sβ11Updated 8 months ago
- β14Updated last year
- β22Updated last year
- QAmeleon introduces synthetic multilingual QA data using PaLM, a 540B large language model. This dataset was generated by prompt tuning Pβ¦β34Updated last year
- **ARCHIVED** Filesystem interface to π€ Hubβ58Updated last year
- β24Updated 2 years ago
- A PyTorch Lightning Callback for pushing models to the Hugging Face Hub π€β‘οΈβ36Updated 2 years ago
- HomebrewNLP in JAX flavour for maintable TPU-Trainingβ49Updated last year
- Serialize JAX, Flax, Haiku, or Objax model params with π€`safetensors`β44Updated 10 months ago
- This repository holds files and scripts for incorporating simple CI/CD practices for model training in ML.β21Updated 3 years ago
- Set of scripts to finetune LLMsβ37Updated last year
- Fast, Modern, Memory Efficient, and Low Precision PyTorch Optimizersβ88Updated 8 months ago
- A miniture AI training framework for PyTorchβ39Updated 2 months ago