Comments (6)
Ok, will add dtype
to __init__
and ensure passed arrays/geometries have that dtype.
from moscot.
amazing, thanks!
from moscot.
I've tried 2 implementations:
- user does
from jax.config import config
config.update("jax_enable_x64", True)
- we check and convert to
f64
- problem is that all allocations (esp. ones not in our control insinkhorn
) are after the update inf64
, which causes a mismatch between sizes.
- add option
x64: bool = False
to fit - this worked better, but it's a stateful operation (I have to write to a config and then NOT revert it for it to work). If previously x64 was disallowed and was temporarily allowed for the computation, we again get some sizing erros (i.e. expected 1600 bytes, got 3200).
Solution would be to convert back tox32
or to remain stateful and I dislike both (not to mention I don't know what the consequences of modifying the config like this are).
My take: if x64 is desired, use as jax
does it in the beginning.
from moscot.
Mhm, that doesn't sound very promising, let's further discuss this today. How can I make sure I use float64
for my internal checks? I would like to run some checks.
from moscot.
Mhm, that doesn't sound very promising, let's further discuss this today. How can I make sure I use float64 for my internal checks? I would like to run some checks.
If you didn't use the above import, it should be always float32
, otherwise float64
.
from moscot.
Assuming this is done.
from moscot.
Related Issues (20)
- test_fgw_problem.py: test_prepare_costs alpha given to prepare HOT 1
- Buggy Test: test_gw_problem gives joint_attr HOT 1
- Buggy Test: `scale_cost` given to prepare in tests/problems/time/test_mixins.py HOT 1
- check the convergence flag
- change links to papers
- adapt low rank gamma default for linear problems HOT 1
- describe birth death process better HOT 1
- Extract the cells in the trace HOT 4
- adapt user guide
- Refactor: Check if both x and x_callback is given
- Refactor: Maybe GW should inherit from FGW instead of vice versa HOT 2
- check for failing tests HOT 1
- check for unrecognized in `solve` methods
- check for warnings thrown by us
- Add Elastic Costs again or asses if they are necessary or not HOT 2
- Plotting Tests Fail HOT 1
- `.impute` consumes too much memory HOT 2
- output the raw cell transition data without aggregation HOT 6
- Full rank initializers
- bug when using tp.compute_interpolated_distance HOT 7
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.
from moscot.