bloom-jax-inference's People
bloom-jax-inference's Issues
Has anyone already tried this repo and verify the results are good?
Hi, I am trying to find BLOOM model in JAX. Thanks for the great work.
Just want to first check if anyone has been running this JAX inference and compare results with PyTorch BLOOM model and verify the results are good?
Thanks
TPU v4 version
Perhaps I overlooked it, but I couldn't find what TPU v4 image your are using for v4-64 pod. Would be a big waste of resources setting up the wrong version ;)
I'm quessing either v2-alpha-tpuv4
or tpu-vm-v4-base
? Would love to know for sure.
Aweseome effort again from HuggingFace and BigScience. Truly amazing
Can this run on NVIDIA-GPU?
As I know, Jax is designed for GOOGLE-TPU, but it can also run on NVIDIA-GPU.
I wonder if jax may be faster or slower than pytorch on GPU for bloom llm inference.
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
๐ Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. ๐๐๐
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google โค๏ธ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.